생각정리/코딩테스트

[JAVA][Level1]PROGRAMMERS 공원 산책

생각중임 2023. 9. 8. 11:31

공원 산책


문제 설명

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

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

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

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

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

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

제한사항

  • 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

입출력 예

park routes result
["SOO","OOO","OOO"] ["E 2","S 2","W 1"] [2, 1]
["SOO","OXX","OOO"] ["E 2","S 2","W 1"] [0, 1]
["OSO","OOO","OXO","OOO"] ["E 2","S 3","W 1"] [0, 0]

입출력 예 설명

입출력 예 #1

입력된 명령대로 동쪽으로 2칸, 남쪽으로 2칸, 서쪽으로 1칸 이동하면 [0,0] -> [0,2] -> [2,2] -> [2,1]이 됩니다.

입출력 예 #2

입력된 명령대로라면 동쪽으로 2칸, 남쪽으로 2칸, 서쪽으로 1칸 이동해야하지만 남쪽으로 2칸 이동할 때 장애물이 있는 칸을 지나기 때문에 해당 명령을 제외한 명령들만 따릅니다. 결과적으로는 [0,0] -> [0,2] -> [0,1]이 됩니다.

입출력 예 #3

처음 입력된 명령은 공원을 나가게 되고 두 번째로 입력된 명령 또한 장애물을 지나가게 되므로 두 입력은 제외한 세 번째 명령만 따르므로 결과는 다음과 같습니다. [0,1] -> [0,0]

주어진 문제

class Solution {
    public int[] solution(String[] park, String[] routes) {
        int[] answer = {};
        return answer;
    }
}

나의 문제풀이

import java.util.ArrayList;
import java.util.List;

class Solution {
    public int[] solution(String[] park, String[] routes) {
        // S와 X의 위치 찾기
        int move_x = 0;
        int move_y = 0;
        List<String> x = new ArrayList<>();
        for (int i = 0; i < park.length; i++) {
            String[] str = park[i].split("");
            for (int j = 0; j < str.length; j++) {
                if (str[j].equals("S")) {
                    move_x = i;
                    move_y = j;
                }
                if (str[j].equals("X")) {
                    x.add("x" + i + "y" + j);
                }
            }
        }

        // 강아지 움직이며 위치 확인
        for (String route : routes) {
            String[] move = route.split(" ");
            boolean check = true;
            int temp_x = move_x;
            int temp_y = move_y;
            switch (move[0]) {
                case "E":
                    for (int i = 0; i < Integer.parseInt(move[1]); i++) {
                        if (x.contains("x" + move_x + "y" + (temp_y + 1))) {
                            check = false;
                            break;
                        }
                        temp_y++;
                    }
                    if (move_y + Integer.parseInt(move[1]) >= park[0].length() || !check) break;
                    move_y += Integer.parseInt(move[1]);
                    break;
                case "W":
                    for (int i = 0; i < Integer.parseInt(move[1]); i++) {
                        if (x.contains("x" + move_x + "y" + (temp_y - 1))) {
                            check = false;
                            break;
                        }
                        temp_y--;
                    }
                    if (move_y - Integer.parseInt(move[1]) < 0 || !check) break;
                    move_y -= Integer.parseInt(move[1]);
                    break;
                case "S":
                    for (int i = 0; i < Integer.parseInt(move[1]); i++) {
                        if (x.contains("x" + (temp_x + 1) + "y" + move_y)) {
                            check = false;
                            break;
                        }
                        temp_x++;
                    }
                    if (move_x + Integer.parseInt(move[1]) >= park.length || !check) break;
                    move_x += Integer.parseInt(move[1]);
                    break;
                case "N":
                    for (int i = 0; i < Integer.parseInt(move[1]); i++) {
                        if (x.contains("x" + (temp_x - 1) + "y" + move_y)) {
                            check = false;
                            break;
                        }
                        temp_x--;
                    }
                    if (move_x - Integer.parseInt(move[1]) < 0 || !check) break;
                    move_x -= Integer.parseInt(move[1]);
                    break;
            }
        }

        return new int[]{move_x, move_y};
    }
}
  • park를 순환하면서
    • 로봇 강아지의 현재 위치 S를 찾아 X, Y 값을 변수에 넣어준다.
    • 장애물 X를 찾아 리스트에 x와 y의 좌표를 문자열로 넣어준다.
  • routes를 순환하면서
  • 방향과 가는 거리를 나눠 switch case에 방향에 따른 이동을 확인한다.
    • 거리를 이동하면서 장애물 X의 리스트에 일치하는 위치를 지나는지
    • 거리를 이동한 거리가 공원 범위 밖인지
    • 위 사항에 해당하지 않으면 거리를 이동한다.

추가한 반례

park routes result
["OOOOOOOOOOO", "OOSOOOOOOOO", "OOOOOOOOOOO", "OOOOOOOOOOO", "OOOOOOOOOOO", "OOOOOOOOOOO", "OOOOOOOOOOO", "OOOOOOOOOOO", "OOOOOOOOOOO", "OOOOOOOOOOO", "OOOOOOOOOOO", "OXOOOOOOOOO"] ["E 9"] [1, 11]
처음에 S와 X를 체크를 하고 주워진 루트대로 이동을 하는데 장애물을 만나면 만나는 위치까지는 이동을 하는 부분으로 이해하고 했었는데 장애물을 만나면 그 이동자체를 안 해 temp를 추가해서 이동하면서 장애물을 만나는 것 을 체크 했다.
또한, X를 리스트로 만들 때 x와 y의 좌표만 넣어서 했을 때, 테스트 케이스 5와 10만 계속 실패를 해서 확인을 해보니 좌표가 1, 11 경우와 11, 1일 경우 둘 다 111로 들어가게 되면서 구분을 할 수 없어 리스트는 이미 contains를 이용하기 위해 문자열로 만들었기 때문에 x와 y 이를 추가해서 구분을 주어 해결했다.

 

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