[Programmers/파이썬] 프로그래머스(Lv.2) 방문 길이 문제풀이

문제 설명

 

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

프로그래머스

 

게임 캐릭터를 4가지 명령어를 통해 움직이려 합니다. 명령어는 다음과 같습니다.

• U: 위쪽으로 한 칸 가기
• D: 아래쪽으로 한 칸 가기
• R: 오른쪽으로 한 칸 가기
• L: 왼쪽으로 한 칸 가기

캐릭터는 좌표평면의 (0, 0) 위치에서 시작합니다. 좌표평면의 경계는 왼쪽 위(-5, 5), 왼쪽 아래(-5, -5), 오른쪽 위(5, 5), 오른쪽 아래(5, -5)로 이루어져 있습니다.

예를 들어, "ULURRDLLU"로 명령했다면

• 1번 명령어부터 7번 명령어까지 다음과 같이 움직입니다.

• 8번 명령어부터 9번 명령어까지 다음과 같이 움직입니다.

이때, 우리는 게임 캐릭터가 지나간 길 중 캐릭터가 처음 걸어본 길의 길이를 구하려고 합니다. 예를 들어 위의 예시에서 게임 캐릭터가 움직인 길이는 9이지만, 캐릭터가 처음 걸어본 길의 길이는 7이 됩니다. (8, 9번 명령어에서 움직인 길은 2, 3번 명령어에서 이미 거쳐 간 길입니다)

단, 좌표평면의 경계를 넘어가는 명령어는 무시합니다.

예를 들어, "LULLLLLLU"로 명령했다면

• 1번 명령어부터 6번 명령어대로 움직인 후, 7, 8번 명령어는 무시합니다. 다시 9번 명령어대로 움직입니다.

이때 캐릭터가 처음 걸어본 길의 길이는 7이 됩니다.

명령어가 매개변수 dirs로 주어질 때, 게임 캐릭터가 처음 걸어본 길의 길이를 구하여 return 하는 solution 함수를 완성해 주세요.

제한사항

• dirs는 string형으로 주어지며, 'U', 'D', 'R', 'L' 이외에 문자는 주어지지 않습니다.
• dirs의 길이는 500 이하의 자연수입니다.

 

사고 흐름

1. 좌표가 나오길래 바로 dfs를 떠올림

2. 문제를 읽어보니 전혀 dfs가 아니라 for문으로 경로 따라가면 될 것 같다고 판단

3. 그리고 방문 확인을 위한 visited 도 만들어줌

 

문제 핵심과 알고리즘

방문 확인을 위한 코드

def solution(dirs):
    answer = 0
    dic = {"U":(0,1), "D":(0,-1), "R":(1,0), "L":(-1,0)} # UDRL
    rev = {"U":"D", "D":"U", "R":"L", "L":"R"}
    visited = []
    now = (0,0)
    
    for di in dirs :
        x, y = now[0], now[1]
        dx, dy = dic[di][0], dic[di][1] 
        if x+dx>5 or x+dx<-5 or y+dy>5 or y+dy<-5 :
            continue
        if (x, y, di) not in visited :
            answer+=1
            visited.append((x, y, di))
            visited.append((x+dx, y+dy, rev[di]))
        now = (x+dx, y+dy)
    
    return answer

 

방향에 따른 좌표 이동을 위한 딕셔너리, 그리고 나중에 필요한 반대방향을 저장한 rev 딕셔너리, 현재 위치인 now를 선언했다. 그리고 이 문제는 dirs 를 for으로 돌면서 범위를 넘는지 확인하고(넘는다면 무시), 만약 방문을 하지 않은 길이라면 방문 표시 하고 now를 갱신한다. 

 

문제 풀면서 헤맸던 부분

처음에는 8번까지만 정답이고 이후로는 오답이였다. 어디에서 허점이였을까 생각하니, 내가 방문처리할 때 처음에는 visited에 (x,y,di) 만 추가했다. 근데 생각해보니 반대에서 오는 것도 길이 겹치는 경우여서 이 또한 방문처리를 했어야 했다. 그래서 visited.append((x+dx, y+dy, rev[di])) 를 추가했다. 

나는 (출발 좌표, 향하는 방향) 으로 방문 여부를 확인했지만, 도착 좌표를 활용하는 등 여러 방법이 있을 것 같다. 

 

총평 

어렿지 않았던 문제였다!