알고리즘

[프로그래머스] 공원산책 (java)

창따오 2024. 1. 9. 11:53
728x90

문제 설명

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

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

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

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

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

 

 

 

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

 
class Solution {
    public int[] solution(String[] park, String[] routes) {
        int x = park.length;
        int y = park[0].length();
        int [] startPos={0,0};
        //동서남북 정의하기
       char [][]map = new char[x][y];
        for(int i = 0 ; i< park.length;i++){
            map[i] = park[i].toCharArray();
            if(park[i].contains("S")){
                startPos[0]=i;
                startPos[1]=park[i].indexOf("S");
            }
        }
        for(String route: routes){
            String []temp = route.split(" ");
            char dir = temp[0].charAt(0);
            int size = Integer.parseInt(temp[1]);
            int nx = startPos[0];
            int ny = startPos[1];
            for(int i=0; i<size;i++){
                if(dir=='E')
                    ny++;
                if(dir=='W')
                    ny--;
                if(dir=='S')
                    nx++;
                if(dir=='N')
                    nx--;
                if(nx<0||nx>=x||ny<0||ny>=y) break;
                if(map[nx][ny]=='X') break;
               if(i== size-1){
                   startPos[0]=nx;
                   startPos[1]=ny;
                   
               }
            }
        }
            
    
        int []answer = {startPos[0],startPos[1]};
        return answer;
    }
}

처음에는 문제를 풀이할 때, 방향을 미리 정의 하였다. 하지만 오히려 코드가 지저분 해졌고, if문을 나열하였지만 가독성을 해치지 않고 오히려 보기 더 좋은 코드가 된 것 같다.

원리는 간단하다.

  • map을 순회하면서 현재 start 지점의 위치정보값을 기억한다.
  • Routes배열을 순회하면서 다음으로 나아갈 방향과 크기를 가지고 와서 갈 수 있다면 현재 위치정보를 옮기면서 기억해준다.