알고리즘

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

winwin-k9 2023. 6. 18. 00:28

문제

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

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

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

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

위 두 가지중 어느 하나라도 해당된다면, 로봇 강아지는 해당 명령을 무시하고 다음 명령을 수행합니다.

공원의 가로 길이가 W, 세로 길이가 H라고 할 때, 공원의 좌측 상단의 좌표는 (0, 0), 우측 하단의 좌표는 (H - 1, W - 1) 입니다.

공원을 나타내는 문자열 배열 park, 로봇 강아지가 수행할 명령이 담긴 문자열 배열 routes가 매개변수로 주어질 때,

로봇 강아지가 모든 명령을 수행 후 놓인 위치를 [세로 방향 좌표, 가로 방향 좌표] 순으로 배열에 담아 return 하도록 solution 함수를 완성해주세요.

풀이

BFS나 DFS를 이용해도 되는데 그러면 너무 복잡해지니 그러진 말자.
Map을 저장하는 배열을 생성해서 저장한 후, start위치를 저장한다.
routes인자들을 뽑아내서 각 NSEW마다 좌표를 보정해주고, 후에 check메소드를 통해서 옳바른 좌표인지, 장애물을 마주쳤는지 체크해준다.
만일 그렇지 못한다면 좌표를 다시 원복시키는데 이를 위해서 값을 상단에서 고정시켜주고 하위에서 원복시켜주는 코드를 구성했다.

 

import java.lang.*;
import java.util.*;

class Solution {
    static String[][] map = null;
    static int x, y;
    public int[] solution(String[] park, String[] routes) {
        map = new String[park.length][park[0].length()];
        
        for(int i = 0; i < park.length; i++) {
            for(int j = 0; j < park[i].length(); j++) {
                map[i][j] = park[i].substring(j, j + 1);
                if (map[i][j].equals("S")) {
                    x = i;
                    y = j;
                }
            }
        }
        
        int subX = x, subY = y;

        for(int i = 0; i < routes.length; i++) {
            String[] str = routes[i].split(" ");
            int way = Integer.parseInt(str[1]);

            for(int j = 0; j < way; j++) {
                switch (str[0]) {
                    case "N":
                        subX -= 1;
                        break;
                    case "S":
                        subX += 1;
                        break;
                    case "W":
                        subY -= 1;
                        break;
                    case "E":
                        subY += 1;
                        break;
                }
                if (!check(subX, subY)) {
                    subX = x;
                    subY = y;
                    break;
                }
            }
            x = subX;
            y = subY;
        }
        int[] answer = {x, y};
        return answer;
    }
    
    static boolean check(int x,int y) {
        return x >= 0 && x < map.length && y >= 0 && y < map[0].length && !map[x][y].equals("X");
    }
}

 

https://github.com/Win-9/Algorism/tree/main/programers/%EA%B3%B5%EC%9B%90%EC%82%B0%EC%B1%85

728x90