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

문제 설명

게임 캐릭터를 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

 


Solution.java

 

처음에는 해당 좌표를 방문했는지 boolean[][] 2차원 배열을 사용하여 풀이하려했습니다.

하지만 2차원 배열로는 테스트를 통과할 수 없었고 3차원 배열을 사용해야 했는데

이 방법보다는 방문한 좌표를 HashSet을 이용해 이미 거쳐간 길은 제외한 채로 추가해

HashSet의 크기를 반환하는 쪽으로 하는 방법이 좀 더 편했습니다.

 

HashSet에 이동한 경로를 추가해야하는데, 이 이동한 경로를 저장한 문자열이 필요합니다.

이때, (0, 0)에서 시작해 위로 올라갔다가 아래로 내려갈 때의 경우 

(0 ,0) -> (0,1) => 0001

(0,1)  -> (0,0) => 0100

이런식으로 문자열을 저장한다면 "이미 거쳐간 길"이지만 중복을 제거 할 수 없습니다.

 

그렇기 때문에,

(0,0)  -> (0,1) => 0001 (currentX + "" + currentY + "->" + nextX +  nextY)

(0,1)  -> (0,0) => 0001 (nextX + "" + nextY + "->" + currentX + currentY)

"이미 거쳐간 길" 중복처리하여 제거할 수 있도록 변수를 저장하도록 해야합니다.

 

import java.util.HashSet;

public class 방문길이 {
    public static void main(String[] args) {
        System.out.println(solution("ULURRDLLU"));
        System.out.println(solution("LULLLLLLU"));
        System.out.println(solution("UDU"));
    }

    public static int solution(String dirs) {
        int answer = 0;

        HashSet<String> visited = new HashSet<>();

        // 시작 위치
        int currentX = 0;
        int currentY = 0;

        for (int i = 0; i < dirs.length(); i++) {
            int nextX = currentX;
            int nextY = currentY;
            String moving = "";

            if (dirs.charAt(i) == 'U') {
                nextY++; // 위로 한칸
                moving += currentX + "" + currentY + "->" + nextX +  nextY;
            } else if (dirs.charAt(i) == 'D') {
                nextY--; // 밑으로 한칸
                moving += nextX + "" + nextY + "->" + currentX + currentY ;
            } else if (dirs.charAt(i) == 'R') {
                nextX++; // 오른쪽으로 한칸
                moving += currentX + "" + currentY + "->" + nextX + nextY;
            } else {
                nextX--; // 왼쪽으로 한칸
                moving += nextX + "" + nextY + "->" + currentX + currentY;
            }

            // 좌표평면의 경계를 넘어가면 추가하지 않고 다음 명령으로 넘어간다.
            if (nextX < -5 || nextY < -5 || nextX > 5 || nextY > 5) {
                continue;
            }
            visited.add(moving);

            currentX = nextX;
            currentY = nextY;
        }
        answer = visited.size();

        return answer;
    }
}