Algorithm/Programmers

[Java] [Level 1] 공원 산책

곽수진 2023. 9. 15. 16:02
반응형

문제

지나다니는 길을 '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

 

 

입출력 예

 

 

입출력 예 설명 #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]

 

 

import java.util.*;
import java.io.*;

class Solution {
    public int[] solution(String[] park, String[] routes) {
        int[] start = getStartPosition(park);
        
        Map<Character, int[]> map = new HashMap<>();
        
        map.put('E', new int[]{0,1});
        map.put('W', new int[]{0,-1});
        map.put('S', new int[]{1,0});
        map.put('N', new int[]{-1,0});
        
        for(String route : routes){
            char direction = route.charAt(0);
            int moveCount = route.charAt(2) - '0';
            
            int[] dxdy = map.get(direction);
            
            int dx = dxdy[0];
            int dy = dxdy[1];
            
            int curX = start[0];
            int curY = start[1];
            
            boolean isOk = true;
            
            for(int i=0; i<moveCount; i++){
                curX += dx;
                curY += dy;
                
                if(curX<0 || curY<0 || curX>park.length-1 || curY>park[0].length()-1){
                    isOk = false;
                    break;
                }
                
                if(park[curX].charAt(curY) == 'X'){
                    isOk = false;
                    break;
                }
            }
            
            if(isOk){
                start[0] = curX;
                start[1] = curY;
            }
        }
        
        return start;
    }
    
    private int[] getStartPosition(String[] park){
        for(int i=0; i<park.length; i++){
            for(int j=0; j<park[i].length(); j++){
                char c = park[i].charAt(j);
                
                if('S' == c){
                    return new int[]{i, j};
                }
            }
        }
        
        return new int[]{0, 0};
    }
}

 

private int[] getStartPosition(String[] park){
    for(int i=0; i<park.length; i++){
        for(int j=0; j<park[i].length(); j++){
            char c = park[i].charAt(j);

            if('S' == c){
                return new int[]{i, j};
            }
        }
    }

    return new int[]{0, 0};
}

▶ 공원을 나타내는 문자열 배열 park를 매개변수로 갖는 getStartPosition 함수를 선언함

▶ 공원의 전체 모습(시작 지점, 이동 가능한 통로, 장애물)을 분석하기 위해 문자 하나 하나를 반복문을 통해 c에 저장함

S인 경우, 즉 시작점인 경우에서부터 시작

  → 시작점을 지정해줘야 함

  → 위 경우에 해당하지 않는 경우 0,0에서 시작함

 

Map<Character, int[]> map = new HashMap<>() : 로봇 강아지의 이동 방향과 이동 거리를 저장할 HashMap을 선언함

 

map.put('E', new int[]{0,1});
map.put('W', new int[]{0,-1});
map.put('S', new int[]{1,0});
map.put('N', new int[]{-1,0});

E, W, S, N은 각각 동, 서, 남, 북을 의미함

  → {세로, 가로}를 의미하기 때문에 동쪽은 가로 1, 서쪽은 가로 -1, 남쪽은 세로 1, 북쪽은 세로 -1을 의미함

 

for(String route : routes){
    char direction = route.charAt(0);
    int moveCount = route.charAt(2) - '0';

    int[] dxdy = map.get(direction);

    int dx = dxdy[0];
    int dy = dxdy[1];

    int curX = start[0];
    int curY = start[1];

    boolean isOk = true;

    for(int i=0; i<moveCount; i++){
        curX += dx;
        curY += dy;

        if(curX<0 || curY<0 || curX>park.length-1 || curY>park[0].length()-1){
            isOk = false;
            break;
        }

        if(park[curX].charAt(curY) == 'X'){
            isOk = false;
            break;
        }
    }

    if(isOk){
        start[0] = curX;
        start[1] = curY;
    }
}
return start;

routes 크기만큼 반복문을 돌리며 값들을 route에 저장함

route 값 중 첫 번째 글짜가 방향을 의미하는 direction에 저장됨

route 값 중 두 번째 글자가 이동 횟수를 의미하는 moveCount에 저장됨

  → 문자 0을 빼는 이유는 route 배열에 이동 횟수가 int가 아닌 String으로 저장되어 있기 때문에 형변환 해줘야 함

이동 횟수만큼 반복문을 돌리며 문제에서 제시된 예외 사항 두 가지를 if문으로 설정하여 boolean 변수인 isOk를 false로 변경함

1. 주어진 방향으로 이동할 때 공원을 벗어나는 경우

2. 주어진 방향으로 이동 중 장애물을 만나는 경우

 

반응형

'Algorithm > Programmers' 카테고리의 다른 글

[Java] [Level 1] 대충 만든 자판  (0) 2023.09.18
[Java] [Level 1] 덧칠하기  (0) 2023.09.17
[Java] [Level 1] 바탕화면 정리  (0) 2023.09.16
[Java] [Level 1] 추억 점수  (1) 2023.09.14
[Java][Level 1] 달리기 경주  (0) 2023.09.13