[프로그래머스] Lv 1. 공원 산책 (Java)

출처: https://school.programmers.co.kr/learn/courses/30/lessons/172928

프로그래머스

---

문제

문제 설명

지나다니는 길을 'O', 장애물을 'X'로 나타낸 직사각형 격자 모양의 공원에서 로봇 강아지가 산책을 하려합니다. 산책은 로봇 강아지에 미리 입력된 명령에 따라 진행하며, 명령은 다음과 같은 형식으로 주어집니다.

• ["방향 거리", "방향 거리" … ]

예를 들어 "E 5"는 로봇 강아지가 현재 위치에서 동쪽으로 5칸 이동했다는 의미입니다. 로봇 강아지는 명령을 수행하기 전에 다음 두 가지를 먼저 확인합니다.

• 주어진 방향으로 이동할 때 공원을 벗어나는지 확인합니다.
• 주어진 방향으로 이동 중 장애물을 만나는지 확인합니다.

위 두 가지중 어느 하나라도 해당된다면, 로봇 강아지는 해당 명령을 무시하고 다음 명령을 수행합니다.
공원의 가로 길이가 W, 세로 길이가 H라고 할 때, 공원의 좌측 상단의 좌표는 (0, 0), 우측 하단의 좌표는 (H - 1, W - 1) 입니다.

공원을 나타내는 문자열 배열 park, 로봇 강아지가 수행할 명령이 담긴 문자열 배열 routes가 매개변수로 주어질 때, 로봇 강아지가 모든 명령을 수행 후 놓인 위치를 [세로 방향 좌표, 가로 방향 좌표] 순으로 배열에 담아 return 하도록 solution 함수를 완성해주세요.

제한사항

• 3 ≤ park의 길이 ≤ 50
  • 3 ≤ park[i]의 길이 ≤ 50
    • park[i]는 다음 문자들로 이루어져 있으며 시작지점은 하나만 주어집니다.
      • S : 시작 지점
      • O : 이동 가능한 통로
      • X : 장애물
  • park는 직사각형 모양입니다.
• 1 ≤ routes의 길이 ≤ 50
  • routes의 각 원소는 로봇 강아지가 수행할 명령어를 나타냅니다.
  • 로봇 강아지는 routes의 첫 번째 원소부터 순서대로 명령을 수행합니다.
  • routes의 원소는 "op n"과 같은 구조로 이루어져 있으며, op는 이동할 방향, n은 이동할 칸의 수를 의미합니다.
    • op는 다음 네 가지중 하나로 이루어져 있습니다.
      • N : 북쪽으로 주어진 칸만큼 이동합니다.
      • S : 남쪽으로 주어진 칸만큼 이동합니다.
      • W : 서쪽으로 주어진 칸만큼 이동합니다.
      • E : 동쪽으로 주어진 칸만큼 이동합니다.
    • 1 ≤ n ≤ 9

입출력 예

---

풀이

로봇 강아지는 명령에 따라 이동하는데 두 가지 상황에서는 해당 명령을 수행하지 않고 다음 명령으로 넘어갑니다.

1. 주어진 방향으로 이동 중 공원을 벗어날 때
2. 주어진 방향으로 이동 중 장애물을 만날 때

두 경우를 코드로 작성하면 다음과 같이 쓸 수 있습니다.

// nx, ny: 주어진 방향으로 이동 중인 좌표
// park: 문제에서 주어진 String 배열 park를 2차원 char 배열로 변환
// isPossible: 이동 가능 여부

// 공원을 벗어날 때
if(nx < 0 || nx >= park.length || ny < 0 || ny >= park[0].length){
    isPossible = false;
}
// 장애물을 만날 때
else{
    if(park[nx][ny] == 'X'){
        isPossible = false;
    }
}

 

명령은 [방향, 거리]로 주어지기 때문에 해당 방향으로 해당 거리만큼 한 칸씩 이동하면서 위 조건을 반복하며 이동 가능 여부를 구합니다. 도중에 불가능한 경우가 생긴다면 isPossible을 false로 하고 반복을 종료하면 됩니다.

 

불가능한 명령은 무시하고 넘어가기 때문에 isPossible이 true일 때만 명령의 방향으로 거리만큼 움직여줍니다.

 

공원을 나타내는 배열이 문자열 배열이기 때문에 확인하기 편하도록 2차원 char 배열로 바꿔주었고, S인 좌표를 시작 좌표로 초기화 합니다.

routes 배열을 하나씩 돌면서 방향과 거리를 구해줍니다. 방향은 EWSN에 해당하는 dx, dy 좌표를 만들어 주고 방향에 맞는 dx, dy의 인덱스를 구합니다.

---

소스코드

class Solution {
    int[] dx = {0, 0, 1, -1};
    int[] dy = {1, -1, 0, 0};
    
    int moveX = -1;
    int moveY = -1;
    
    public int[] solution(String[] park, String[] routes) {
        int[] answer = new int[2];
        char[][] arr = new char[park.length][park[0].length()];
        
        for(int i=0; i<park.length; i++){
            for(int j=0; j<park[0].length(); j++){
                arr[i][j] = park[i].charAt(j);
                
                if(arr[i][j] == 'S'){
                    moveX = i;
                    moveY = j;
                }
            }
        }
        
        for(String route : routes){
            String direction = route.split(" ")[0];
            int num = Integer.parseInt(route.split(" ")[1]);
            int dir = -1;
            
            if(direction.equals("E")) dir = 0;
            else if(direction.equals("W")) dir = 1;
            else if(direction.equals("S")) dir = 2;
            else if(direction.equals("N")) dir = 3;
            
            move(arr, dir, num);
        }
        
        answer[0] = moveX;
        answer[1] = moveY;
        
        return answer;
    }
    
    public void move(char[][] park, int dir, int num){
        int nx = moveX, ny = moveY;
        boolean isPossible = true;
        
        for(int i=0; i<num; i++){
            nx += dx[dir];
            ny += dy[dir];
            
            if(nx < 0 || nx >= park.length || ny < 0 || ny >= park[0].length){
                isPossible = false;
                break;
            }
            else{
                if(park[nx][ny] == 'X'){
                    isPossible = false;
                    break;
                }
            }
            
        }
        
        if(isPossible){
            moveX += dx[dir] * num;
            moveY += dy[dir] * num;
        }
    }
    
}