[프로그래머스] 개인정보 수집 유효기간 - 자바(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

입출력 예

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) {
        char[][] arr = new char[park.length][park[0].length()];
        int x = 0;
        int y = 0;

        for (int i = 0; i < park.length; i++) {
            // 공원을 2차원 배열로 변환
            arr[i] = park[i].toCharArray();

            // 시작 위치 좌표 찾기
            if (park[i].contains("S")) {
                x = park[i].indexOf('S');
                y = i;
            }
        }

        // routes 순회
        for (String route : routes) {
            // 이동 방향 (E, W, S, N)
            String direction = route.split(" ")[0];
            // 이동 거리
            int distance = Integer.parseInt(route.split(" ")[1]);
            int nx = x;
            int ny = y;

            // 주어진 방향의 거리만큼 이동
            for (int i = 0; i < distance; i++) {
                switch (direction) {
                    case "E" -> nx++;
                    case "W" -> nx--;
                    case "S" -> ny++;
                    default -> ny--;
                }

                // 새로운 좌표가 공원 배열의 범위 내에 있을 경우
                if (nx >= 0 && ny >= 0 && ny < arr.length && nx < arr[0].length) {
                    // 장애물을 만났을 경우 이동 중단
                    if (arr[ny][nx] == 'X') {
                        break;
                    }

                    // 이동 거리만큼 이동한 경우 좌표 업데이트
                    if (i == distance - 1) {
                        x = nx;
                        y = ny;
                    }
                }
            }
        }

        return new int[]{y, x};
    }
}

코드 설명

주석 참고

• 이 문제는 공원과 로봇 강아지의 이동 경로를 통해 최종 위치 좌표를 찾아 반환하는 문제다.
• 먼저, park 1차원 배열을 2차원 배열로 변환하기 위한 char 타입의 2차원 배열 arr을 선언하고, x, y 좌표를 0으로 초기화한다.
• park 배열을 순회하여 2차원 배열 arr로 변환하면서 시작 좌표를 찾는다.
  • 2차원 배열로 변환 : toCharArray() 메서드를 통해 문자열을 1차원 배열로 분할하여 arr[i]에 담는다.
  • 시작 좌표 찾기 : park[i]의 문자열에 S가 포함된 경우 해당 index를 x 좌표, i를 y좌표로 변경한다.
• 이후 routes 배열을 순회하여 로봇 강아지의 이동 경로를 처리한다.
  • route 문자열의 공백(" ")을 기준으로 split()하여 이동 방향(direction)과 거리(distance)를 초기화한다.
  • 이동에 따른 좌표를 기록할 ny와 ny를 각각 x, y로 초기화한다.
  • distance 만큼 반복하여 direction에 따라 좌표를 1씩 증가시킨다.
  • 증가시킬 때마다 해당 좌표가 공원 배열의 범위를 벗어나는지 판단한다.
  • 만약, 증가시킨 좌표에 장애물("X")이 존재하는 경우 반복문을 중단한다.
  • 이후 distance 만큼 반복한 경우 x, y를 기록한 nx, ny로 업데이트한다.