따봉도치야 고마워

[프로그래머스] 방문 길이 본문

프로그래밍/알고리즘

[프로그래머스] 방문 길이

따봉도치 2020. 5. 27. 21:14

문제 설명

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

 


풀이

- 일단 '좌표'의 방문 여부를 체크하는게 아니라, '길'의 방문 여부를 체크하는 것

- (0.1) -> (0.2) 와 (0.2) -> (0.1)은 같은 길을 의미 함.

 

1. 4차원 배열 사용

import java.lang.Math;
class Solution {
    public int solution(String dirs) {
        int answer = 0;     
        int x = 0, y = 0;
        int [][][][]check = new int[11][11][11][11];
        for(Character dir : dirs.toCharArray()){
            int nextX = x, nextY = y;

            if(dir == 'U') nextY++;
            if(dir == 'D') nextY--;
            if(dir == 'R') nextX++;
            if(dir == 'L') nextX--;   

            if(Math.abs(nextX) > 5 || Math.abs(nextY) > 5){
                continue;
            }

            if(check[nextX+5][nextY+5][x+5][y+5] != 1){
                check[nextX+5][nextY+5][x+5][y+5] = 1;
                check[x+5][y+5][nextX+5][nextY+5] = 1;
                answer ++;
            }
            
            x = nextX; y = nextY;
        }

        return answer;
    }
}

- 배열 인덱스에 +5를 해줘 마이너스 좌표 처리

- 길을 처음 방문할 때, 출발지와 목적지를 바꿔서 양방향 전부 1로 체크해줌

 

2. HashSet 이용

- HashSet : 중복을 허용하지 않는 집합

import java.util.HashSet;

class Solution {
    public int solution(String dirs) {
        int answer = 0;     
        int x = 0, y = 0;
        
        HashSet<String> set = new HashSet<>();
        
        for(Character dir : dirs.toCharArray()){
            int lastX = x, lastY = y;
            
            if(dir == 'U' && y+1 <= 5) y++;
            if(dir == 'D' && y-1 >= -5) y--;
            if(dir == 'R' && x+1 <= 5) x++;
            if(dir == 'L' && x-1 >= -5) x--;   
            
            if(lastX == x && lastY == y){
                continue;
            }
            
            //양방향 다 넣어주기
            set.add(lastX+""+lastY+""+x+""+y);
            set.add(x+""+y+""+lastX+""+lastY);
        }
        
        answer = set.size()/2;
        return answer;
    }
}

- HashSet을 만들고 방문할 때 마다 해당 길의 양방향 정보를 다 넣어줌

- 어차피 중복 값은 들어가지 않기 때문에 싹 다 넣고 마지막에 set.size()/2로 계산

 

2-2. HashMap을 이용해 계산 간략화

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

class Solution {
    public int solution(String dirs) {
        int answer = 0;     
        int x = 0, y = 0;
        
        HashSet<String> set = new HashSet<>();
        HashMap<Character, int[]> map = new HashMap<>();
        
        map.put('U',new int[]{0,1});
        map.put('D',new int[]{0,-1});
        map.put('R',new int[]{1,0});
        map.put('L',new int[]{-1,0});
        
        for(Character dir : dirs.toCharArray()){
            int lastX = x, lastY = y;
            
            x += map.get(dir)[0];
            y += map.get(dir)[1];
            
            if(x>5 || x<-5){
                x -= map.get(dir)[0];
                continue;
            }
            
            if(y>5 || y<-5){
                y -= map.get(dir)[1];
                continue;
            }
                      
            //양방향 다 넣어주기
            set.add(lastX+""+lastY+""+x+""+y);
            set.add(x+""+y+""+lastX+""+lastY);
        }
        
        answer = set.size()/2;
        return answer;
    }
}

- HashMap에 각 명령어에 맞는 좌표 이동 수를 넣어놓고 계산 시 Map에서 꺼내서 계산

Comments