[Baekjoon] 1261번 알고스팟

문제

알고스팟 운영진이 모두 미로에 갇혔다. 미로는 N*M 크기이며, 총 1*1크기의 방으로 이루어져 있다. 미로는 빈 방 또는 벽으로 이루어져 있고, 빈 방은 자유롭게 다닐 수 있지만, 벽은 부수지 않으면 이동할 수 없다.

알고스팟 운영진은 여러명이지만, 항상 모두 같은 방에 있어야 한다. 즉, 여러 명이 다른 방에 있을 수는 없다. 어떤 방에서 이동할 수 있는 방은 상하좌우로 인접한 빈 방이다. 즉, 현재 운영진이 (x, y)에 있을 때, 이동할 수 있는 방은 (x+1, y), (x, y+1), (x-1, y), (x, y-1) 이다. 단, 미로의 밖으로 이동 할 수는 없다.

벽은 평소에는 이동할 수 없지만, 알고스팟의 무기 AOJ를 이용해 벽을 부수어 버릴 수 있다. 벽을 부수면, 빈 방과 동일한 방으로 변한다.

만약 이 문제가 알고스팟에 있다면, 운영진들은 궁극의 무기 sudo를 이용해 벽을 한 번에 다 없애버릴 수 있지만, 안타깝게도 이 문제는 Baekjoon Online Judge에 수록되어 있기 때문에, sudo를 사용할 수 없다.

현재 (1, 1)에 있는 알고스팟 운영진이 (N, M)으로 이동하려면 벽을 최소 몇 개 부수어야 하는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 미로의 크기를 나타내는 가로 크기 M, 세로 크기 N (1 ≤ N, M ≤ 100)이 주어진다. 다음 N개의 줄에는 미로의 상태를 나타내는 숫자 0과 1이 주어진다. 0은 빈 방을 의미하고, 1은 벽을 의미한다.

(1, 1)과 (N, M)은 항상 뚫려있다.

출력

첫째 줄에 알고스팟 운영진이 (N, M)으로 이동하기 위해 벽을 최소 몇 개 부수어야 하는지 출력한다.

예제

Example 1:

Input: 
3 3
011
111
110
Output: 
3

Example 2:

Input:
4 2
0001
1000
Output:
0

Example 3:

Input:
6 6
001111
010000
001111
110001
011010
100010
Output:
2

조건

시간 제한 : 1초

메모리 제한 : 128 MB

풀이과정

내 풀이

화성탐사문제와 유사한 문제이다. distance를 2차원 리스트로 생성하고 다익스트라 알고리즘을 활용하여 계산한다.

import heapq

# 무한대 정의
INF = int(1e9)

# m = 가로 크기, n = 세로 크기
m, n = map(int, input().split())

# 미로 정보 입력 받기
graph = []
for i in range(n):
    graph.append(list(map(int, input())))
    
# 최단 거리 테이블을 모두 무한으로 초기화
distance = [[INF] * m for _ in range(n)]

# 시작 위치는 (0, 0)
x, y = 0, 0
q = [(graph[x][y], x, y)]
distance[x][y] = graph[x][y]

# 위치 이동
dx = [0, 1, 0, -1]
dy = [1, 0, -1, 0]

# 다익스트라 알고리즘 수행
while q:
    dist, x, y = heapq.heappop(q)
    if distance[x][y] < dist:
        continue
    for i in range(4):
        nx = x + dx[i]
        ny = y + dy[i]
        if nx <= -1 or ny <= -1 or nx >= n or ny >= m:
            continue
        cost = dist + graph[nx][ny]
        # 현재 노드를 거쳐서, 다른 노드로 이동하는 거리가 더 짧은 경우
        if cost < distance[nx][ny]:
            distance[nx][ny] = cost
            heapq.heappush(q, (cost, nx, ny))
            
# 값 출력
print(distance[n - 1][m - 1])