[프로그래머스] 보물 지도 (Python)

프로그래머스 문제 풀이 - PCCP 모의고사 #2

📰 문제

문제 확인 🏃

💡 입출력 예제

4, 4, [[2, 3], [3, 3]]

>> 5

5, 4, [[1, 4], [2, 1], [2, 2], [2, 3], [2, 4], [3, 3], [4, 1], [4, 3], [5, 3]]

>> -1

💬 풀이

🎵 첫번째 풀이

from collections import deque

def solution(col, row, hole):
    
    graph = [[0] * col for _ in range(row)]
    for c, r in hole:   
        graph[abs(r-row)][c-1] = -1   # 함정 표시  

    visited = [[[0] * 2 for _ in range(col)] for _ in range(row)]
    visited[row-1][0][1] = True

    q = deque()
    q.append((row-1, 0, 1))

    dr = [-1, 1, 0, 0]
    dc = [0, 0, 1, -1]
    
    answer = 0

    while q:

        for _ in range(len(q)):
            r, c, flag = q.popleft()

            if r == 0 and c == col-1:
                return answer

            for d in range(4):
                nr = r + dr[d]
                nc = c + dc[d]

                if 0 <= nr < row and 0 <= nc < col and not graph[nr][nc] and not visited[nr][nc][flag]:
                    visited[nr][nc][flag] = True
                    q.append((nr, nc, flag))
                
                if flag:
                    nr += dr[d]
                    nc += dc[d]
                    if 0 <= nr < row and 0 <= nc < col and not graph[nr][nc] and not visited[nr][nc][0]:
                        visited[nr][nc][0] = True
                        q.append((nr, nc, 0))

        answer += 1

    return -1

✒️ 생각

문제 자체에서 주어진 좌표 방법이 있지만, 헷갈려서 그냥 내가 원래 풀던 식으로 좌표 변경해서 풀었음!
3차원은 아직 익숙하지가 않아..