[프로그래머스 Lv.1] 공원 산책 | Java

바다·2024년 5월 17일

Java Lv.1 TIL 구현 코딩테스트 프로그래머스

코딩테스트

목록 보기

1/10

문제 설명

문제 바로가기 클릭! 🖱️

지나다니는 길을 '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. 시작 위치 파악 & 위치 배열 만들기

int[] answer = {0, 0};
int h = park.length;
int w = park[0].length();

//위치를 저장할 배열 선언
char[][] parks = new char[h][w];
        
for(int i = 0; i<h; i++) {
	for(int j = 0; j<w; j++) {
   		parks[i][j] = park[i].charAt(j);	//위치 저장
        if(parks[i][j] == 'S') {			//시작 위치라면, answer에 시작 위치 저장
           	answer[0] = i;
            answer[1] = j;
    	}       
	}
}

2. 이동을 위한 배열 & 방향 파악을 위한 map 만들기

int[] dh = {-1, 0, 1, 0};
int[] dw = {0, 1, 0, -1};

Map<Character, Integer> map = new HashMap<>();
map.put('N', 0);
map.put('E', 1);
map.put('S', 2);
map.put('W', 3);

3. routes 배열을 돌며, 공원 산책

	for(int i = 0; i<routes.length; i++) {
            char direction = routes[i].charAt(0);	//방향
            int value = Integer.parseInt(routes[i].substring(2, routes[i].length()));	//값
            
            int index = map.get(direction);		//map에서 위치에 알맞은 index 찾기
            
            boolean check = true;	//이동해도 되는지 체크할 boolean 값
            
            int nh = answer[0];		//검사에 쓰일 temp 값
            int nw = answer[1];
            
            for(int j = 0; j<value; j++) {	//이동할 값만큼 반복
                nh += dh[index];
                nw += dw[index];
                
                if(nh >= h || nh < 0 || nw >= w || nw < 0) {
                	//공원을 벗어난다면 다음 명령 시행
                    check = false;
                    break;
                }
                if(parks[nh][nw] == 'X') {
                	//이동하는 길에 장애물이 있다면 다음 명령 시행
                    check = false;
                    break;
                }
            }
            
            //이동해도 되면 answer의 값 변경
            if(check) {
                answer[0] = nh;
                answer[1] = nw;
            }
            
        }

전체 코드

import java.util.*;

class Solution {
    public int[] solution(String[] park, String[] routes) {
        int[] answer = {0, 0};
        int h = park.length;
        int w = park[0].length();
        
        char[][] parks = new char[h][w];
        
        for(int i = 0; i<h; i++) {
            for(int j = 0; j<w; j++) {
                parks[i][j] = park[i].charAt(j);
                
                if(parks[i][j] == 'S') {
                    answer[0] = i;
                    answer[1] = j;
                } 
            }
        }
        
        int[] dh = {-1, 0, 1, 0};
        int[] dw = {0, 1, 0, -1};
        
        Map<Character, Integer> map = new HashMap<>();
        map.put('N', 0);
        map.put('E', 1);
        map.put('S', 2);
        map.put('W', 3);
        
        for(int i = 0; i<routes.length; i++) {
            char direction = routes[i].charAt(0);
            int value = Integer.parseInt(routes[i].substring(2, routes[i].length()));
            
            int index = map.get(direction);
            
            boolean check = true;
            
            int nh = answer[0];
            int nw = answer[1];
            
            for(int j = 0; j<value; j++) {
                nh += dh[index];
                nw += dw[index];
                
                if(nh >= h || nh < 0 || nw >= w || nw < 0) {
                    check = false;
                    break;
                }
                if(parks[nh][nw] == 'X') {
                    check = false;
                    break;
                }
            }
            
            if(check) {
                answer[0] = nh;
                answer[1] = nw;
            }
            
        }

        return answer;
    }
}