Summer/Winter Coding(~2018)에 나온 방문 길이 문제입니다.
문제
방문 길이
게임 캐릭터를 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번 명령어에서 이미 거쳐 간 길입니다)
단, 좌표평면의 경계를 넘어가는 명령어는 무시합니다.
제한 사항
- dirs는 string형으로 주어지며, 'U', 'D', 'R', 'L' 이외에 문자는 주어지지 않습니다.
- dirs의 길이는 500 이하의 자연수입니다.
문제 해설
캐릭터의 위치를 설정하고(5, 5) 캐릭터가 움직일때 지나가는 길을 저장해서 현재 지나가는 길이 이미 지나갔던 길인지 확인하면 된다. 제한 사항에 dirs의 길이는 500 이하라고 하니 원하는 대로 저장해도, 메모리가 초과될 걱정은 하지 않아도 된다. 주의 할 점은 캐릭터가 L((5, 5) -> (4, 5)) R((4, 5) -> (5, 5))처럼 움직인다면 L로 지나갔던 길을 R로 되돌아 왔으므로 지나간 길의 갯수는 1개가 된다. 즉 길은 방향성이 없는 양방향 길이다.
def solution(dirs):
answer = 0
roads = set()
position = (5, 5)
next_position = (0, 0)
dirs = list(dirs)
for w in dirs:
if w == "U":
next_position = (position[0], max(0, position[1] - 1))
elif w == "D":
next_position = (position[0], min(10, position[1] + 1))
elif w == "R":
next_position = (min(10, position[0] + 1), position[1])
elif w == "L":
next_position = (max(0, position[0] - 1), position[1])
if next_position != position and (next_position, position) not in roads:
roads.add((position, next_position))
roads.add((next_position, position))
answer +=1
position = next_position
return answer좌표 평면을 벗어난다면 움직이지 않게 되니, 현재 위치와 다음 위치가 같게 된다. 그렇다면 아무 길도 지나지 않게 된다.
움직이기 전 위치와 움직인 후의 위치를 roads에 저장한다. 저장 할 때 LR과 같은 경우에 대응할 수 있게 양방향을 저장한다.
처음에 list로 짯는데 list를 이용해서 값을 저장하다 보니 deepcopy를 이용하게 됐고, 현재 set을 이용한 것 보다 느렸다. 문제에서는 효율성을 체크하지 않아서 그냥 통과가 됐지만 set과 list를 적절한 상황에 이용하는게 좋다고 생각해 코드를 변경했다.
