알고리즘 문제 풀이/항해99 코테 스터디

[99클럽/파이썬 챌린저/14일차] 미로 탈출 명령어

제유찬 2024. 11. 11. 00:46

프로그래머스 미로 탈출 명령어 LV3

 

문제
n x m 격자 미로가 주어집니다. 당신은 미로의 (x, y)에서 출발해 (r, c)로 이동해서 탈출해야 합니다.
단, 미로를 탈출하는 조건이 세 가지 있습니다.
격자의 바깥으로는 나갈 수 없습니다.(x, y)에서 (r, c)까지 이동하는 거리가 총 k여야 합니다. 이때, (x, y)와 (r, c) 격자를 포함해, 같은 격자를 두 번 이상 방문해도 됩니다. 미로에서 탈출한 경로를 문자열로 나타냈을 때, 문자열이 사전 순으로 가장 빠른 경로로 탈출해야 합니다.
이동 경로는 다음과 같이 문자열로 바꿀 수 있습니다.
l: 왼쪽으로 한 칸 이동 r: 오른쪽으로 한 칸 이동 u: 위쪽으로 한 칸 이동 d: 아래쪽으로 한 칸 이동
예를 들어, 왼쪽으로 한 칸, 위로 한 칸, 왼쪽으로 한 칸 움직였다면, 문자열 "lul"로 나타낼 수 있습니다.
미로에서는 인접한 상, 하, 좌, 우 격자로 한 칸씩 이동할 수 있습니다.
격자의 크기를 뜻하는 정수 n, m, 출발 위치를 뜻하는 정수 x, y, 탈출 지점을 뜻하는 정수 r, c, 탈출까지 이동해야 하는 거리를 뜻하는 정수 k가 매개변수로 주어집니다. 이때, 미로를 탈출하기 위한 경로를 return 하도록 solution 함수를 완성해 주세요. 단, 위 조건대로 미로를 탈출할 수 없는 경우 "impossible"을 return 해야 합니다.


제한사항
2 ≤ n (= 미로의 세로 길이) ≤ 50
2 ≤ m (= 미로의 가로길이) ≤ 50
1 ≤ x ≤ n, 1 ≤ y ≤ m, 1 ≤ r ≤ n, 1 ≤ c ≤ m, (x, y) ≠ (r, c)
1 ≤ k ≤ 2,500


입출력 예
n m x y r c k result
3 4 2 3 3 1 5 "dllrl"
2 2 1 1 2 2 2 "dr"
3 3 1 2 3 3 4 "impossible"

 

 

현재 이동 가능한 정점에서 down, left, right, up 순서로 k 번간 모두 탐색하였다.

최단 경로를 찾은 후에 남은 시간을 사용하는 방법을 쓰려고 하였으나 시간 초과에  따로 걸리지 않아 넘어갔다.

 

더보기
from collections import deque

def solution(n, m, x, y, r, c, k):
    move_list = [[1,0,'d'],[0,-1,'l'],[0,1,'r'],[-1,0,'u']]
    limit_need = abs(x-r) + abs(y-c)
    if limit_need > k or (k-limit_need) % 2 != 0: return "impossible"
    
    visit = [[[""] * m for _ in range(n)] for __ in range(2)]
    dq = deque()
    dq.append((x-1, y-1))

    for t in range(k):
        visit[(t+1)%2] = [[""] * m for _ in range(n)]
        for _ in range(len(dq)):
            i, j = dq.popleft()
            for ay, ax, di in move_list:
                dy, dx = i+ay, j+ax
                if dy<0 or dy>=n or dx<0 or dx>=m: continue
                if visit[(t+1)%2][dy][dx] != "": continue
                visit[(t+1)%2][dy][dx] = visit[t%2][i][j] + di
                dq.append((dy, dx))

    return visit[k%2][r-1][c-1]​