알고리즘

게임 맵 최단거리 - 프로그래머스 Python

침착하고 가야할 곳에만 집중하는 달팽이 2026. 2. 18. 13:48
문제: 
ROR 게임은 두 팀으로 나누어서 진행하며, 상대 팀 진영을 먼저 파괴하면 이기는 게임입니다. 따라서, 각 팀은 상대 팀 진영에 최대한 빨리 도착하는 것이 유리합니다.
격자로 maps 이 주어지며 0이면 지나갈 수 없는 벽, 1이면 갈 수 있는 길입니다. 항상 0,0 에서 출발해서 n,m에 있는 상대 팀 진영에 가야 합니다. 
게임 맵의 상태 maps가 매개변수로 주어질 때, 캐릭터가 상대 팀 진영에 도착하기 위해서 지나가야 하는 칸의 개수의 최솟값을 return 하도록 solution 함수를 완성해주세요. 단, 상대 팀 진영에 도착할 수 없을 때는 -1을 return 해주세요.
제한 사항:
maps는 n x m 크기의 게임 맵의 상태가 들어있는 2차원 배열로, n과 m은 각각 1 이상 100 이하의 자연수입니다.
n과 m은 서로 같을 수도, 다를 수도 있지만, n과 m이 모두 1인 경우는 입력으로 주어지지 않습니다.
maps는 0과 1로만 이루어져 있으며, 0은 벽이 있는 자리, 1은 벽이 없는 자리를 나타냅니다.
처음에 캐릭터는 게임 맵의 좌측 상단인 (1, 1) 위치에 있으며, 상대방 진영은 게임 맵의 우측 하단인 (n, m) 위치에 있습니다.

 

 

생각해 볼 점 
1. 생각할 것도 없다. BFS 임.
2. 최대한 빨리 도착해라? BFS임. 

 

내 풀이: 

from collections import deque
def solution(maps):
    n = len(maps)
    m = len(maps[0])
    
    dr = [-1, 1, 0, 0]
    dc = [0,0,-1,1]
    
    visited = [[0]*m for _ in range(n)]
    visited[0][0] = 1
    
    que = deque()
    que.append((0,0))
    
    while que:
        r,c = que.popleft()
        
        if r == n-1 and c == m-1:
            return visited[n-1][m-1]
            
        for i in range(4):
            nr = r + dr[i]
            nc = c + dc[i]
            
            if nr >= n or nc >= m or nr < 0 or nc < 0 or maps[nr][nc] == 0 or visited[nr][nc] != 0:
                continue
            
            que.append((nr,nc))
            visited[nr][nc] = visited[r][c] + 1
    return -1

도착했는지 확인하는 코드를 언제 넣어야 할지 좀 고민했다... 

 if r == n-1 and c == m-1:

 

popleft 를 하고 지금 있는 칸이 도착칸인지 확인해야 하나

vs

nr,nc 인 다음 칸이 도착칸인지 확인해야 하나? 

나는 다음 칸이 도착칸인지 확인하는게 더 효율적이라고 생각했는데 그렇게 하면 시작칸 = 도착칸일 때는 체크를 못한다. (물론 이 문제에서는 1*1인 격자는 안 나온다고 되어 있어서 문제 없긴 하다) 

결론 => 안전하게 popleft 한 다음에 현재 칸이 도착칸인지 확인하자!