프로그래머스 - 공원 산책(Java)

KDG: First things first!·2024년 8월 7일
0

프로그래머스

목록 보기
6/18

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/172928?language=java




문제 설명

지나다니는 길을 '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[] answer = new int[2];

        int H = park.length; // 세로
        int W = park[0].length(); // 가로

        int x = -1; // 시작 X 좌표
        int y = -1; // 시작 Y 좌표


        for (int i = 0; i < park.length; i++) {
            if (park[i].contains("S")) {
                x = i;
                y = park[i].indexOf('S');
                break;
            }
        }


        for (String route : routes) {
            String[] array = route.split(" "); // 방향, 거리
            String dir = array[0]; // 이동 방향
            int dis = Integer.parseInt(array[1]); // 이동 거리
            Integer currentX = x; // 현재 X 좌표
            Integer currentY = y; // 현재 Y 좌표
            boolean flag = true; // 이동 가능 여부

            // 이동 방향에 따른 이동 처리 -> 동, 서로 이동은 Y 좌표가 변경되고 남, 북으로 이동은 X 좌표가 변경됨
            switch (dir) {
                case "E": // 동쪽
                    for (int i = 0; i < dis; i++) { // 이동 거리만큼 반복
                        currentY++; // 동쪽으로 1 이동
                        if (currentY >= W || park[currentX].charAt(currentY) == 'X') { // 범위를 벗어나거나 장애물이 있을 경우
                            flag = false; // 이동 불가 판정으로 변경
                            break;
                        }
                    }
                    if (flag) y = currentY; // 이동 가능할 경우 움직인 위치 반영해서 현재 Y 좌표 변경
                    break; // 반복문 탈출

                case "W": // 서쪽
                    for (int i = 0; i < dis; i++) { // 이동 거리만큼 반복
                        currentY--; // 서쪽으로 1 이동
                        if (currentY < 0 || park[currentX].charAt(currentY) == 'X') { // 범위를 벗어나거나 장애물이 있을 경우
                            flag = false; // 이동 불가 판정으로 변경
                            break;
                        }
                    }
                    if (flag) y = currentY; // 이동 가능할 경우 움직인 위치 반영해서 현재 Y 좌표 변경
                    break;

                case "S": // 남쪽
                    for (int i = 0; i < dis; i++) { // 이동 거리만큼 반복
                        currentX++; // 남쪽으로 1 이동
                        if (currentX >= H || park[currentX].charAt(currentY) == 'X') { // 범위를 벗어나거나 장애물이 있을 경우
                            flag = false; // 이동 불가 판정으로 변경
                            break;
                        }
                    }
                    if (flag) x = currentX; // 이동 가능할 경우 움직인 위치 반영해서 현재 X 좌표 변경
                    break;

                case "N": // 북쪽
                    for (int i = 0; i < dis; i++) { // 이동 거리만큼 반복
                        currentX--; // 북쪽으로 1 이동
                        if (currentX < 0 || park[currentX].charAt(currentY) == 'X') { // 범위를 벗어나거나 장애물이 있을 경우
                            flag = false; // 이동 불가 판정으로 변경
                            break;
                        }
                    }
                    if (flag) x = currentX; // 이동 가능할 경우 움직인 위치 반영해서 현재 X 좌표 변경
                    break;

            }

        }
        answer[0] = x; // 최종 X 좌표
        answer[1] = y; // 최종 Y 좌표

        return answer;

    }
}
profile
알고리즘, 자료구조 블로그: https://gyun97.github.io/

0개의 댓글