본문 바로가기
알고리즘

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

by 창따오 2024. 1. 9.
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배열을 순회하면서 다음으로 나아갈 방향과 크기를 가지고 와서 갈 수 있다면 현재 위치정보를 옮기면서 기억해준다.