백준 온라인 저지, BFS / 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)으로 이동하기 위해 벽을 최소 몇 개 부수어야 하는지 출력한다.

제한

예제 입력 1

3 3
011
111
110

예제 출력 1

3

예제 입력 2

4 2
0001
1000

예제 출력 2

0

예제 입력 3

6 6
001111
010000
001111
110001
011010
100010

예제 출력 3

2

힌트

이 문제는 알고스팟에서도 풀 수 있다.

접근 방법

- 0. 주어진 위치를 입력받는다.
- 1. 현재 위치에서 상하좌우 중 이동할 수 있는 위치를 확인하여 1을 거치면 1을 거친 횟수에 1을 더한 뒤, [1을 거친 횟수, 현재 위치]의 형태로 min_heap에 저장한다.
- 2. 이후 min_heap에서 값을 하나씩 삭제하며 1번을 반복한다.
- 3. 상하좌우의 위치를 탐색하다 (n, m)의 위치가 나오면 탐색을 멈추고 1을 거친 횟수를 출력한다.

코드

# 접근 방법
# 0. 주어진 위치를 입력받는다.
# 1. 현재 위치에서 상하좌우 중 이동할 수 있는 위치를 확인하여 1을 거치면 1을 거친 횟수에 1을 더한 뒤, [1을 거친 횟수, 현재 위치]의 형태로 min_heap에 저장한다.
# 2. 이후 min_heap에서 값을 하나씩 삭제하며 1번을 반복한다.
# 3. 상하좌우의 위치를 탐색하다 (n, m)의 위치가 나오면 탐색을 멈추고 1을 거친 횟수를 출력한다. 
import sys, heapq
input = sys.stdin.readline
m, n = map(int, input().split())
board = [[int(x) for x in input().rstrip()] for _ in range(n)]
visited = [[False for _ in range(m)] for _ in range(n)]

def move(row, col):
    q = []
    heapq.heappush(q, [0, row, col])
    
    while q:
        count, r, c = heapq.heappop(q)
        if visited[r][c]:
            continue
            
        visited[r][c] = True
        for dr, dc in [[1, 0], [-1, 0], [0, 1], [0, -1]]:
            if 0<=r+dr<=n-1 and 0<=c+dc<=m-1 and not visited[r+dr][c+dc]:
                if board[r+dr][c+dc]:
                    heapq.heappush(q, [count+1, r+dr, c+dc])
                else:
                    if r+dr == n - 1 and c+dc == m - 1:
                        return count
                    heapq.heappush(q, [count, r+dr, c+dc])
    return 0

result = move(0, 0)
print(result)