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

문제

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

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

풀이

선분을 set에 넣으면 되겠다, 라는 생각으로 풀었다.
선분의 시작점과 끝점을 set에 넣는데, 양방향이므로 두 개의 방향을 다 넣어준다.
두개의 방향을 다 넣었으니 정답으로는 나누기 2를 해주면 된다.

def solution(dirs):
    answer = set()
    cur = (0, 0)
    for i in dirs:
        y, x = cur
        if i == 'U':
            y += 1
        elif i == 'D':
            y -= 1
        elif i == 'R':
            x += 1
        elif i == 'L':
            x -= 1
        if -5 <= y and y <= 5 and -5 <= x and x <= 5:
            nxt = (y, x)
            answer.add((nxt, cur)) 
            answer.add((cur, nxt))
            cur = nxt
    return len(answer) // 2