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

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/49994

프로그래머스

 

문제 설명

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

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

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

이 때, 게임 캐릭터가 지나간 길 중 캐릭터가 처음 걸어본 길의 길이를 구하려고 한다.

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

 

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

제한사항

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

 

풀이

가는 길이 중복되거나 경계 밖의 길의 처리는 어렵지 않지만, 갔던 길을 다시 되돌아오는 길일 경우를 처리해야 하기 때문에 생각을 했던 문제였다.

만약, (0, 0) 에서 'R'로 (1, 0)으로 갔을 때, (1, 0) 에서 (0, 0) 으로 가는 길도 처음 걸어본 길이 아니기 때문에, 자동으로 처리해주어야 한다.

 

✅ Code - 성공

def solution(dirs):
    answer = set()
    
    arr = [[0] * 11 for _ in range(11)]
    sx, sy = 5, 5
    for dir in dirs:
        dx, dy = sx, sy
        if dir == "U":
            dx, dy = dx - 1, dy
        elif dir == "L":
            dx, dy = dx, dy - 1
        elif dir == "R":
            dx, dy = dx, dy + 1
        elif dir == "D":
            dx, dy = dx + 1, dy
        
        if 0 <= dx <= 10 and 0 <= dy <= 10:
            answer.add((sx, sy, dx, dy))
            answer.add((dx, dy, sx, sy))
            sx, sy = dx, dy
            
    return len(answer) // 2