LV2 방문길이

---

문제

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

프로그래머스

풀이

 

이동한 길만 체크하는 문제이다.

 

이렇게 저렇게 삽질좀 해보다가, 두 지점의 좌표를 set에 넣고 검사해보면 되겠다 싶었다.

 

(1, 0) => (1, 1) 이라면 set에 (1, 0, 1, 1)과 같이 넣는다. 단, (1, 1) => (1, 0) 도 같은 경로를 사용하기에 여기에 대한 검사를 해줘야한다. 따라서 (1, 0) => (1, 1) 을 갈때 (1, 0, 1, 1) 과 (1, 1, 1, 0) 이 set에 없을때만 set에 넣어준다.

 

그리고 범위안에 들어오면 set에 추가하는 여부와 상관없이 좌표는 갱신해주면 된다.

 

코드

def solution(dirs):
    
    dic = {'U':(-1, 0), 'D':(1, 0), 'R':(0, 1), 'L':(0, -1)}
    x = y = 0
    s = set()    
    for d in dirs:
        nx = x + dic[d][0]
        ny = y + dic[d][1]
    
        if -5<=nx<=5 and -5<=ny<=5:
            if (x, y, nx, ny) not in s and (nx, ny, x, y) not in s:
                s.add((x, y, nx, ny))
            x = nx
            y = ny
            
    return len(s)

 

후기

 

나처럼 하지않고 set에 (1, 0, 1, 1), (1, 1, 1, 0) 모두 넣고 set사이즈/2 하는 방법도 있더라.