[programmers/py] 게임 맵 최단거리

승민·2024년 1월 24일

알고리즘

목록 보기
48/171

게임 맵 최단거리

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

문제 설명

위 그림에서 검은색 부분은 벽으로 막혀있어 갈 수 없는 길이며, 흰색 부분은 갈 수 있는 길입니다. 캐릭터가 움직일 때는 동, 서, 남, 북 방향으로 한 칸씩 이동하며, 게임 맵을 벗어난 길은 갈 수 없습니다.

maps는 n x m 크기의 게임 맵의 상태가 들어있는 2차원 배열로, n과 m은 각각 1 이상 100 이하의 자연수입니다.
n과 m은 서로 같을 수도, 다를 수도 있지만, n과 m이 모두 1인 경우는 입력으로 주어지지 않습니다.
maps는 0과 1로만 이루어져 있으며, 0은 벽이 있는 자리, 1은 벽이 없는 자리를 나타냅니다.
처음에 캐릭터는 게임 맵의 좌측 상단인 (1, 1) 위치에 있으며, 상대방 진영은 게임 맵의 우측 하단인 (n, m) 위치에 있습니다.

게임 맵의 상태 maps가 매개변수로 주어질 때, 캐릭터가 상대 팀 진영에 도착하기 위해서 지나가야 하는 칸의 개수의 최솟값을 return

단, 상대 팀 진영에 도착할 수 없을 때는 -1을 return 해주세요.

설명

deque에 지나갈 수 있는 경우를 보관
visited에 최소 길이를 넣는다.

visited를 여러번 방문할 수 있어서 -1이 아니면 못 들어가게 막아
가장 빨리 도착하는 경우에만 값이 변경되게 했다.

from collections import deque

def solution(maps):  
    # 이동
    dx = [0, 0, 1, -1]
    dy = [1, -1, 0, 0]
    
    row = len(maps)
    col = len(maps[0])
    
    visited = [[-1]*col for _ in maps]
    
    dq = deque([(0,0,1)])
    visited[0][0] = 1

    while dq:
        x, y, m = dq.popleft()
                
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            
            if 0 <= nx < row and 0 <= ny < col and maps[nx][ny] == 1:
                if visited[nx][ny] == -1:
                    if nx == row-1 and ny == col -1:
                        return m+1
                    dq.append((nx, ny, m+1))
                    visited[nx][ny] = m+1
                    
    return -1

0개의 댓글