[Programmers] 방문 길이

 

 

 

해설 코드 

 

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

분석 > set 을 활용해 요소를 추가하는 법을 배웠다. set을 활용하면 같은 길을 걸었는지, 아닌지를 분별할 수 있다. 길은 양방향(현위치와 전위치가 반전되어야 같은 길을 걸은 것)이기 때문에 x와 y를 각각 왼쪽, 오른쪽으로 나누어 계산해주어야 한다. 한쪽에만 몰아서 계산해준다면, set 배열이 중복을 거르지 못하기 때문이다. 

 

 

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