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

문제주소 :programmers.co.kr/learn/courses/30/lessons/49994#

코딩테스트 연습 - 방문 길이

---

<문제 설명>

문제 설명

게임 캐릭터를 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
문제의 예시와 같습니다.

 

<풀이법>

▒ 한줄 개념: 해시맵(디렉토리) ▒ 

해시맵(디렉토리)를 이용하여 풀면 쉬운 문제입니다. 

처음에는 각 좌표에 대한 방문여부를 확인하면 될 것이라고 생각했는데, 아니었습니다.

하나의 좌표마다 4방향의 진입로가 있으므로 4방향의 진입로를 다 따져줘야합니다.

또한 한 좌표에서 다른 좌표로 이동할 때 이동 전 좌표에서 나가는 것과 이동 후 좌표에서 들어오는 것, 두 가지를 모두 체크해주어야 합니다. (2,1) / 'D' 를 통해서 (1,1)로 내려가는 경로와 (1,1) / 'U' 을 통해서 (2,1)로 올라가는 경로는 같기 때문입니다.

 

• 모든 좌표계에 대한 해시맵 생성 (history{})
• 이동 전 좌표와 이동 후 좌표에 대한 이동 값이 해시 맵에 이미 있는지 확인
  • 해시맵에 존재하지 않는 이동일 경우 추가 후 answer+1
    • ex) (1,1)에서 'U'가 입력되었을 경우, 이동 전((1,1):'D' 과 이동 후 (1,2):'U') 둘 다 저장
    • (1,1)/'U' -> history[(1,1)].append('D') / history[(1,2)].append('U')
• 최종 answer 값 반환

 

 

 

<코드(Python)>

def solution(dirs):
    answer = 0
    history = {}
    for i in range(-5, 6):
        for j in range(-5, 6):
            history[(i,j)] = []
            
    x = 0
    y = 0
    for d in dirs:
        prev = (x,y)
        
        if d == 'L' and x != -5:
            prev_d = 'R'
            x -= 1
        elif d == 'R' and x != 5:
            prev_d = 'L'
            x += 1
        elif d == 'U' and y != 5:
            prev_d = 'D'
            y += 1
        elif d == 'D' and y != -5:
            prev_d = 'U'
            y -= 1
        else:
            continue
         
        if d not in history[(x,y)] and prev_d not in history[prev]:
            history[(x,y)].append(d)
            history[prev].append(prev_d)
            answer += 1
                
        
    return answer

 

 

더 많은 코드 보기(GitHub) : github.com/dwkim-97/CodingTest