99클럽 코테 스터디 27일차 TIL (구현)

JAVA / Lv 0. 공원 산책

 

문제

공원을 나타내는 문자열 배열 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

입출력 예

---

문제 해결

• 시작 좌표 설정: 완전탐색으로 S인 문자열을 찾고 인덱스를 저장하는 함수 생성
• 좌표 이동 : routers 배열 순회하며 lx, ly 변수 값 변경
  이동한 값이 0이상이어야하고, 배열의 범위를 벗어나면 안된다.
  그리고 장애물이 있으면 현재 이동은 취소되고, 다음 routes[i]를 처리합니다.
  장애물이 없고, 모든 이동을 성공적으로 완료하면, start 위치를 갱신

나의 풀이

import java.util.*;

class Solution {
    public int[] solution(String[] park, String[] routes) {
        // 시작 지점
        int[] start = getStartPosition(park); // [0.0]
        
        for(int i = 0; i < routes.length; i++){
            String op = routes[i].split(" ")[0]; // E
            int mv = Integer.parseInt(routes[i].split(" ")[1]); // 2
            
            int lx = start[0];
            int ly = start[1];
            
            for(int j = 0; j < mv; j++){
                if(op.equals("N")){ // 북쪽
                    lx--;
                }
                if(op.equals("S")){ // 남쪽
                    lx++;
                }
                
                if(op.equals("E")){ // 동쪽
                    ly++;
                }
                if(op.equals("W")){ // 서쪽
                    ly--;
                }
                
                if(lx >= 0 && ly >= 0 && lx < park.length && ly < park[0].length()){
                    if(park[lx].charAt(ly) == 'X'){
                        break;
                    }
                    
                    if(j == mv - 1){
                        start[0] = lx;
                        start[1] = ly;
                    }
                }
            }
        }
        
        return start;
    }
    
    public int[] getStartPosition(String[] park){
        for(int i = 0; i < park.length; i++){
            for(int j = 0; j < park[i].length(); j++){
                if(park[i].charAt(j) == 'S'){
                    return new int[]{i, j};
                }
            }
        }
        
        return null;
    }
    
}

 

---