본문 바로가기

개발지/코테

[프로그래머스] 코테연습 - 공원 산책(lv.1)

- 내 방법 풀이

  • 레벨 1의 문제였고, 조건도 간단했지만 생각보다 코드로 구현해내기가 복잡했던 것 같다.
  • 동 서 남 북 4방향의 선택지가 있었고, 방향에 따라 행동제약에 걸어줘야할 조건문이 달라졌기 때문에 중간중간 헷갈려서 몇 번이나 코드를 고쳐야 했다.
  • 처음 생각한 방법으로 코드를 구현해 냈으며, 복잡할 땐 그림을 그리고 예시를 따져가며 썼다.
  • route를 실행할 수 없다면 다음 route를 실행해야 하기 때문에, 먼저 boolean으로 이동할 수 있는지 검사하는 메서드를 만들었고, 인수에 따라 다른 조건문을 따라 가도록 했다.
  • 이후 해당 route를 실행할 수 있다면 로봇 위치를 조정해 주었는데, 알고 보니 최초 로봇 위치가 park배열 안에 S로 주어졌다. 그래서 최초 위치를 검사하는 로직을 추가한 뒤 마무리 했다.
  • 간단해 보이지만 그렇지 않았던.. 그래서 시간이 오래 걸렸던 문제였다.
class Solution {
    public int[] solution(String[] park, String[] routes) {
        int[] answer = new int[2];
        
        for (int i = 0; i < park.length; i ++) { // 현재 위치 찾기
            for (int j = 0; j < park[0].length(); j++) {
                if (park[i].charAt(j) == 'S') {
                    answer[0] = i;
                    answer[1] = j;
                    break;
                }
            }
        }
        
        for (String route : routes) { // 이동경로만큼 순회
            char op = route.charAt(0); // 방향
            int n = route.charAt(2) - '0'; // 이동거리
            
            if (canGo(park, answer, op, n)) { // 이동할 수 있다면
                switch (op) {
                    case 'N' : answer[0] -= n; 
                        break;
                    case 'E' : answer[1] += n;
                        break;
                    case 'S' : answer[0] += n;
                        break;
                    case 'W' : answer[1] -= n;
                        break;
                }
            }
        }
        return answer;
    }
    
    public boolean canGo(String[] map, int[] S, char op, int n) {
        boolean can = true;
        
        if (op == 'N') { // 북으로 이동
            if (S[0] - n < 0) return false; // 이동했을 때 맵을 벗어나는지 검사
            
            for (int i = S[0] - n; i <= S[0]; i++) { // 이동 경로 중 장애물 있는지 검사
                if (map[i].charAt(S[1]) == 'X') {
                    return false;
                }
            }
        } else if (op == 'E') { // 동으로 이동
            if (S[1] + n > map[0].length() - 1) return false;
            
            for (int i = S[1]; i <= S[1] + n; i++) {
                if (map[S[0]].charAt(i) == 'X') {
                    return false;
                }
            }
        } else if (op == 'S') { // 남으로 이동 
            if (S[0] + n > map.length - 1) return false;
            
            for (int i = S[0]; i <= S[0] + n; i++) {
                if (map[i].charAt(S[1]) == 'X') {
                    return false;
                }
            }
        } else { // 서로 이동
            if (S[1] - n < 0) return false;
            
            for (int i = S[1] - n; i <= S[1]; i++) {
                if (map[S[0]].charAt(i) == 'X') {
                    return false;
                }
            }
        }
        return can;
    }
}

- 다른 사람 방법 풀이

  • boolean isWalk를 사용해, 현재위치에서 가려는 방향을 getdirectionIndex메서드로 구해준다.
  • distance거리만큼 이동하면서 공원을 벗어나는지, 혹은 장애물 구간에 있는지를 확인한다.
  • 확인 후 getdirectionIndex를 통해 구한 방향으로 이동한다. 이 때 int[][]arr 에서 방향에 따라 알맞은 배열을 가져와 기존 위치 좌표인 x,y 배열에 더해준다
class Solution {
     public static int[] solution(String[] park, String[] routes) {
        int x = 0;
        int y = 0;
        int[][] arr = {{-1,0},{1,0},{0,-1},{0,1}};
        boolean start = false;
        for(int i=0; i< park.length;i++){
            if(start){
                break;
            }
            for(int j=0; j<park[i].length();j++){
                if(park[i].charAt(j) == 'S'){
                    x = i;
                    y = j;
                    start = true;
                    break;
                }
            }
        }

        for (String route : routes) {
            String[] routeArr = route.split(" ");
            String direction = routeArr[0];
            int distance = Integer.parseInt(routeArr[1]);

            int index = getDirectionIndex(direction);

            if(isWalk(park, x, y, distance, arr[index])){
                x += distance*arr[index][0];
                y += distance*arr[index][1];
            }
        }

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

    private static boolean isWalk(String[] park, int x, int y, int distance, int[] arr) {
        for(int i = 0; i< distance; i++){
            x+= arr[0];
            y+= arr[1];

            if(y<0 || y> park[0].length()-1 || x<0 || x> park.length-1 || park[x].charAt(y)=='X'){
                return false;
            }
        }
        return true;
    }

    private static int getDirectionIndex(String direction) {
        int index = 0;
        switch (direction){
            case "N":
                break;
            case "S":
                index = 1;
                break;
            case "W":
                index = 2;
                break;
            case "E":
                index = 3;
                break;
        }
        return index;
    }
}