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

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

💻   코드

def solution(dirs):
    result = [0, 0]
    route = {}
    dir_dict = {
        "U": [0, 1],
        "D": [0, -1],
        "L": [-1, 0],
        "R": [1, 0],
    }

    for dir in dirs:
        if (-5 <= result[0] + dir_dict[dir][0] <= 5) and (-5 <= result[1] + dir_dict[dir][1] <= 5):
            new_x = result[0] + dir_dict[dir][0]
            new_y = result[1] + dir_dict[dir][1]

            route[str(result[0]) + str(result[1]) + str(new_x) + str(new_y)] = 1
            route[str(new_x) + str(new_y) + str(result[0]) + str(result[1])] = 1

            result[0] = new_x
            result[1] = new_y

    return len(route) // 2

 

 

✅   풀이

이 문제의 핵심은

1) 이미 왔던 경로를 또 카운팅 하지 않는 것,

2) 한 경로에 대해 왔던 방향과 갔던 방향을 동일하게 카운팅 할 것

 

파이썬에서 동일한 값을 하나로 카운팅 하는 자료형식은 딕셔너리와 set 가 대표적이다.

그렇기 때문에 route 라는 딕셔너리를 만들어 하나의 경로에 대해 두가지 방향(오는 방향, 가는 방향) 을 저장한다. 

이렇게 하면 동일한 경로를  몇번이나 왔다 갔다하더라도 1회로 카운팅 한다.

 

 

사실상 dictionary 가 아닌 set 을 사용하는 것이 더 옳은 접근인 것 같다. 리팩토링 해봐야겠다.