BAEKJOON #1261 알고스팟 (BFS, 다익스트라) -python

nathan·2021년 11월 5일

알고리즘문제

목록 보기

83/102

알고스팟

시간 제한	메모리 제한
1초	128MB

문제

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

입출력 예시

예제 입력 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

풀이

생각

최단경로를 구하는 문제이다.
BFS, 다익스트라를 이용하여 풀 수 있다.
최소힙을 이용하여 (충돌횟수, x좌표, y좌표)를 원소로 담아 목적지까지 가는 경우를 찾았다.
최소힙을 이용한 이유는 충돌횟수가 최소가 되는 선에서 목적지까지 도달하는 것을 체크하기 위함이었다.
visited를 통해 방문 여부를 체크하고, 0이면 그대로 힙에 다시 넣어주고 1이라면 crash에 1을 추가하여 힙에 넣어주는 방식으로 진행하였다.

python code

# 백준 1261번 알고스팟
from sys import stdin
from collections import deque
from heapq import heappop, heappush, heapify
input = stdin.readline

n, m = map(int, input().split())
matrix = []
for i in range(m):
    matrix.append(input().rstrip())

visited= [[False]*n for _ in range(m)]
# print(matrix)

# q = deque([(0,0,0)]) #(r, c, crash)
q = [(0, 0, 0)] # (crash, r, c)
heapify(q)
visited[0][0] = True
# 동, 남, 서, 북
dx = [0, +1, 0, -1]
dy = [+1, 0, -1, 0]
while q:
    crash, r, c = heappop(q)
    if r == n-1 and c == m-1:
        print(crash)
        break
    for i in range(4):
        new_r = r + dx[i]
        new_c = c + dy[i]
        if 0 <= new_r < n and 0 <= new_c < m:   # 범위 확인
            if visited[new_c][new_r] == False:  # 방문 여부 확인
                visited[new_c][new_r] = True
                if matrix[new_c][new_r] == "0":
                    heappush(q, (crash, new_r, new_c))
                else:
                    heappush(q, (crash+1, new_r, new_c))