[프로그래머스] 게임 맵 최단거리 - python 본문

코테 문제 풀이

[프로그래머스] 게임 맵 최단거리 - python

미니모아 2022. 4. 18. 18:25
반응형

게임 맵 최단거리

문제

게임 맵의 상태 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) 위치에 있습니다.

풀이

dfs로 풀었더니 효율성에서 런타임 에러가 나서 bfs로 풀었다.

bfs를 돌면서 이전 값을 현재 위치에 더해준다.

마지막 도착점의 값이 1보다 크면 해당 값을 리턴하고 작으면 -1을 리턴한다.

from collections import deque
def solution(maps): 
    n = len(maps)
    m = len(maps[0])
    
    directions = [(1, 0), (-1, 0), (0, -1), (0, 1)]
    q = deque([(0, 0)])
    
    while q:
        cx, cy = q.popleft()
        for dx, dy in directions:
            nx = cx + dx
            ny = cy + dy 
            if 0 <= nx < n and 0 <= ny < m and maps[nx][ny] == 1:
                maps[nx][ny] += maps[cx][cy]
                q.append((nx, ny))
    
    
    return maps[n-1][m-1] if maps[n-1][m-1] > 1 else -1
반응형
Comments