[백준] 1251. 알고스팟(Python, 파이썬)

백준 온라인의 문제 풀이입니다.

📖 [백준] 1251. 알고스팟

시간 제한	메모리 제한	난이도
1초	128 MB	골드 4

알고스팟 운영진이 모두 미로에 갇혔다. 미로는 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)으로 이동하기 위해 벽을 최소 몇 개 부수어야 하는지 출력한다.

✔️ 입력 출력 예시

입력	출력
3 3 011 111 110	3
4 2 0001 1000	0
6 6 001111 010000 001111 110001 011010 100010	2

🗒️ 내 풀이

해당 문제는 BFS로 풀 수 있는데 벽울 부수는 횟수를 저장하는 리스트를 따로 선언해야하며, 빈 방인 경우 횟수가 증가하지 않아 queue에 appendleft를 통해 우선순위를 두어 먼저 탐색하도록 한다.

import sys
from collections import deque
input = sys.stdin.readline

m, n = map(int, input().split())
graph = [list(map(int, input().strip())) for _ in range(n)]
dx = [1, 0, -1, 0]
dy = [0, 1, 0, -1]

dist = [[-1] * m for _ in range(n)] # 벽을 부수는 횟수 리스트

def bfs():
    q = deque([(0, 0)])
    dist[0][0] = 0
    while q:
        x, y = q.popleft()
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            if 0 <= nx < n and 0 <= ny < m and dist[nx][ny] == -1:
                if graph[nx][ny] == 0: # 빈 방인 경우
                    # appendleft로 빈 방인 경우 우선순위를 둠
                    q.appendleft((nx, ny))
                    dist[nx][ny] = dist[x][y]
                else: # 벽인 경우
                    q.append((nx, ny))
                    # 벽을 부수는 횟수 + 1
                    dist[nx][ny] = dist[x][y] + 1

bfs()
print(dist[n - 1][m - 1])

사실 위의 개념이 다익스트라 알고리즘과 동일하므로 다익스트라 알고리즘으로 풀 수 있다.

import sys
import heapq
input = sys.stdin.readline
INF = 1e9
m, n = map(int, input().split())
graph = [list(map(int, input().strip())) for _ in range(n)]
dx = [1, 0, -1, 0]
dy = [0, 1, 0, -1]
distance = [[INF] * m for _ in range(n)]

def dijkstra():
    q = []
    heapq.heappush(q, (0, 0, 0))
    distance[0][0] = 0
    while q:
        dist, x, y = heapq.heappop(q)
        if dist > distance[x][y]:
            continue
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            if 0 <= nx < n and 0 <= ny < m:
                cost = dist + graph[nx][ny]
                if cost < distance[nx][ny]:
                    distance[nx][ny] = cost
                    heapq.heappush(q, (cost, nx, ny))

dijkstra()
print(distance[n - 1][m - 1])