[BOJ] 1261 알고스팟 (Python3)

문제

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

알고스팟 운영진은 여러명이지만, 항상 모두 같은 방에 있어야 한다. 즉, 여러 명이 다른 방에 있을 수는 없다. 어떤 방에서 이동할 수 있는 방은 상하좌우로 인접한 빈 방이다. 즉, 현재 운영진이 (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)으로 이동하기 위해 벽을 최소 몇 개 부수어야 하는지 출력한다.

from collections import deque
m, n = map(int, input().split())

graph = []

for i in range(n):
    temp = []
    
    for j in input():
        temp.append(int(j))
        
    graph.append(temp)
    
INF = 987654321 
    
blockGraph = [[INF for _ in range(m)] for _ in range(n)]

q = deque()

q.append([0, 0, 0])
blockGraph[0][0] = 0

dx = [1, -1, 0, 0]
dy = [0, 0, 1, -1]

while q:
    x, y, block = q.popleft()
    
    for i in range(4):
        
        nx = x + dx[i]
        ny = y + dy[i]
        
        if 0 <= nx < n and 0 <= ny < m:
            if graph[nx][ny] == 1:
                if block + 1 < blockGraph[nx][ny]:
                    blockGraph[nx][ny] = min(blockGraph[nx][ny], block + 1)
                    q.append([nx, ny, block + 1])
                else:
                    pass
            else:
                if block < blockGraph[nx][ny]:
                    blockGraph[nx][ny] = min(blockGraph[nx][ny], block)
                    q.append([nx, ny, block])
                    
print(blockGraph[n - 1][m - 1])
               

 3 3
 011
 110
 011


3

문제 풀이

bfs를 활용할줄 알면 바로 풀 수 있는 문제다.

x - 1, x + 1, y - 1, y + 1을 이동해가며 값을 기록한다? -> 바로 bfs

단, 이 문제는 보통의 bfs처럼 벽을 피해 최단거리의 이동경로를 구하는게 아니라, 거리는 상관없이 무조건 최소로 벽을 부수는 이동 경로를 구하는 문제다.

따라서 부순 벽의 최솟값을 기록하는 그래프를 하나 더 만들어준다. 나의 경우는 blockGraph라는 이름으로 만들어줬다. bfs를 순차적으로 진행해가며, 현재 경로의 blockGraph에 기록된 값보다 부순벽의 수가 적은 경우에만 값을 갱신하고 이를 큐에 넣어주면 된다.