[Programmers] 방문 길이 바로가기
게임 캐릭터를 4가지 명령어를 통해 움직이려 합니다. 명령어는 다음과 같습니다.
U: 위쪽으로 한 칸 가기
D: 아래쪽으로 한 칸 가기
R: 오른쪽으로 한 칸 가기
L: 왼쪽으로 한 칸 가기
캐릭터는 좌표평면의 (0, 0) 위치에서 시작합니다. 좌표평면의 경계는 왼쪽 위(-5, 5), 왼쪽 아래(-5, -5), 오른쪽 위(5, 5), 오른쪽 아래(5, -5)로 이루어져 있습니다.
예를 들어, "ULURRDLLU"로 명령했다면
이때, 우리는 게임 캐릭터가 지나간 길 중 캐릭터가 처음 걸어본 길의 길이를 구하려고 합니다. 예를 들어 위의 예시에서 게임 캐릭터가 움직인 길이는 9이지만, 캐릭터가 처음 걸어본 길의 길이는 7이 됩니다. (8, 9번 명령어에서 움직인 길은 2, 3번 명령어에서 이미 거쳐 간 길입니다)
단, 좌표평면의 경계를 넘어가는 명령어는 무시합니다.
예를 들어, "LULLLLLLU"로 명령했다면
이때 캐릭터가 처음 걸어본 길의 길이는 7이 됩니다.
명령어가 매개변수 dirs로 주어질 때, 게임 캐릭터가 처음 걸어본 길의 길이를 구하여 return 하는 solution 함수를 완성해 주세요.
dirs | answer |
---|---|
"ULURRDLLU" | 7 |
"LULLLLLLU" | 7 |
입출력 예 #1
문제의 예시와 같습니다.
입출력 예 #2
문제의 예시와 같습니다.
dirs
에 따라 이동하다가 좌표 평면의 경계에서 벗어나는 경우는 continue
를 통해 명령어를 무시한다.set()
)을 통해 현재 이동한 길이 초행인지 아닌지를 확인하여 answer
의 값에 1을 더해준다.✍ 코드
directions = {'U' : (0,-1), 'D': (0,1), 'R': (1,0), 'L':(-1,0)}
def solution(dirs):
answer = 0
S = set()
x, y = 0, 0
for dir in dirs:
dx, dy = directions[dir]
nx, ny = x + dx, y + dy
if nx < -5 or nx > 5 or ny < -5 or ny > 5: continue
if (x,y,nx,ny) not in S and (nx,ny,x,y) not in S:
answer += 1
S.add((x,y,nx,ny))
S.add((nx,ny,x,y))
x, y = nx, ny
return answer