[Python] 방문 길이

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

문제보기

풀이

1. 델타 사용. 

2. 가는 '길'이 중복되는 것은 제외 -> set 사용

3. set에 저장할 때, 좌표점이 아닌 가는 길을 저장.

 

def solution(dirs):
    dir = {'U':(1,0), 'D':(-1,0), 'L':(0,-1), 'R':(0,1)}
    road = set()
    cr, cc = 0, 0
    
    for d in dirs:
        nr, nc = cr + dir[d][0], cc + dir[d][1]
        if -5 <= nr <= 5 and -5 <= nc <= 5:
            road.add((cr, cc, nr, nc))
            road.add((nr, nc, cr, cc))
            cr, cc = nr, nc
    
    return len(road) // 2