[프로그래머스] 방문길이 - 자바(Java)

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

프로그래머스

 

문제 설명

게임 캐릭터를 4가지 명령어를 통해 움직이려 합니다. 명령어는 다음과 같습니다.

• U: 위쪽으로 한 칸 가기
• D: 아래쪽으로 한 칸 가기
• R: 오른쪽으로 한 칸 가기
• L: 왼쪽으로 한 칸 가기

캐릭터는 좌표평면의 (0, 0) 위치에서 시작합니다. 좌표평면의 경계는 왼쪽 위(-5, 5), 왼쪽 아래(-5, -5), 오른쪽 위(5, 5), 오른쪽 아래(5, -5)로 이루어져 있습니다.

 

예를 들어, "ULURRDLLU"로 명령했다면

 

• 1번 명령어부터 7번 명령어까지 다음과 같이 움직입니다.

 

• 8번 명령어부터 9번 명령어까지 다음과 같이 움직입니다.

 

이때, 우리는 게임 캐릭터가 지나간 길 중 캐릭터가 처음 걸어본 길의 길이를 구하려고 합니다. 예를 들어 위의 예시에서 게임 캐릭터가 움직인 길이는 9이지만, 캐릭터가 처음 걸어본 길의 길이는 7이 됩니다. (8, 9번 명령어에서 움직인 길은 2, 3번 명령어에서 이미 거쳐 간 길입니다)

단, 좌표평면의 경계를 넘어가는 명령어는 무시합니다.

예를 들어, "LULLLLLLU"로 명령했다면

• 1번 명령어부터 6번 명령어대로 움직인 후, 7, 8번 명령어는 무시합니다. 다시 9번 명령어대로 움직입니다.

 

이때 캐릭터가 처음 걸어본 길의 길이는 7이 됩니다.

명령어가 매개변수 dirs로 주어질 때, 게임 캐릭터가 처음 걸어본 길의 길이를 구하여 return 하는 solution 함수를 완성해 주세요.

 

제한사항

• dirs는 string형으로 주어지며, 'U', 'D', 'R', 'L' 이외에 문자는 주어지지 않습니다.
• dirs의 길이는 500 이하의 자연수입니다.

입출력 예

dirs	answer
"ULURRDLLU"	7
"LULLLLLLU"	7

입출력 예 설명

입출력 예 #1
문제의 예시와 같습니다.

입출력 예 #2
문제의 예시와 같습니다.

 

코드

import java.util.HashMap;
import java.util.HashSet;

class Solution {
    // 1. 좌표평면을 벗어나는지 체크
    private static boolean isValidMove(int nx, int ny) {
        return 0 <= nx && nx < 11 && 0 <= ny && ny < 11;
    }
    
    // 2. 다음 좌표 결정을 위한 HashMap 생성
    private static final HashMap<Character, int[]> location = new HashMap<>();
    
    private static void initLocation() {
        location.put('U', new int[]{0, 1});
        location.put('D', new int[]{0, -1});
        location.put('L', new int[]{-1, 0});
        location.put('R', new int[]{1, 0});
    }
    
    public int solution(String dirs) {
        initLocation();
        int x = 5, y = 5;
        // 3. 겹치는 좌표는 1개로 처리하기 위함
        HashSet<String> answer = new HashSet<>();
        // 4. 주어진 명령어로 움직이면서 좌표 저장
        for (int i=0; i<dirs.length(); i++) {
            int[] offset = location.get(dirs.charAt(i));
            int nx = x + offset[0];
            int ny = y + offset[1];
            // 5. 벗어난 좌표는 인정하지 않음
            if(!isValidMove(nx, ny))
                continue;
            // 6. A에서 B로 간 경우 B에서 A도 추가해야 함(총 경로의 개수는 방향성이 없음)
            answer.add(x + " " + y + " " + nx + " " + ny);
            answer.add(nx + " " + ny + " " + x + " " + y);
            // 7. 좌표를 이동했으므로 업데이트
            x = nx;
            y = ny;
        }
        return answer.size() / 2;
    }
}

 

정리

1. 중복 경로는 최종 길이에 포함하지 않는다. 중복을 포함하지 않는다는 문장이 나오면 HashSet을 생각하면 좋습니다.

2. 음수 좌표를 포함합니다. (좌표 범위 -5 <= x, y <= 5) 좌표는 방향에 의해서만 제어됩니다. 따라서 원점을 (0, 0)에서 (5, 5)로 바꿔 음수 좌표 문제를 해결해도 됩니다.

3. answer.add(x + " " + y + " " + nx + " " + ny) ➡️ (x, y)에서 (nx, ny)까지의 경로를 방문했다 라고 기록합니다.

이때 (x,y)에서 (nx, ny)만 저장하는 것이 아니라 그 반대인 (nx, ny)에서 (x, y) 역시 추가합니다.

4. N은 dirs의 길이입니다. dirs 길이만큼 순회하므로 시간 복잡도는 O(N) 입니다.