[프로그래머스] 방문 길이 - python

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

프로그래머스

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

다른 분 풀이를 참고 했는데 이미 방문한곳을 집합으로 처리하면서 간단하게 해결 가능하다.

 

set 을 이용하고 add를 이용하자