(Python) - BOJ(2665번) : 미로만들기

https://www.acmicpc.net/problem/2665

2665번: 미로만들기

---

• 문제 : 

n×n 바둑판 모양으로 총 n2개의 방이 있다. 일부분은 검은 방이고 나머지는 모두 흰 방이다. 검은 방은 사면이 벽으로 싸여 있어 들어갈 수 없다. 서로 붙어 있는 두 개의 흰 방 사이에는 문이 있어서 지나다닐 수 있다. 윗줄 맨 왼쪽 방은 시작방으로서 항상 흰 방이고, 아랫줄 맨 오른쪽 방은 끝방으로서 역시 흰 방이다.

시작방에서 출발하여 길을 찾아서 끝방으로 가는 것이 목적인데, 아래 그림의 경우에는 시작방에서 끝 방으로 갈 수가 없다. 부득이 검은 방 몇 개를 흰 방으로 바꾸어야 하는데 되도록 적은 수의 방의 색을 바꾸고 싶다.

아래 그림은 n=8인 경우의 한 예이다.

위 그림에서는 두 개의 검은 방(예를 들어 (4,4)의 방과 (7,8)의 방)을 흰 방으로 바꾸면, 시작방에서 끝방으로 갈 수 있지만, 어느 검은 방 하나만을 흰 방으로 바꾸어서는 불가능하다. 검은 방에서 흰 방으로 바꾸어야 할 최소의 수를 구하는 프로그램을 작성하시오.

단, 검은 방을 하나도 흰방으로 바꾸지 않아도 되는 경우는 0이 답이다.

 

 

 

---

• 풀이 :

인접행렬로 표현된 그래프에서의 탐색이며, 0-1 BFS로 해결할 수 있다.

0-1 BFS는 간선의 가중치가 0 또는 1인 그래프에서 BFS를 수행하는 것이며, 이는 queue가 아닌 heap을 사용하여 쉽게 구현할 수 있다.

가중치가 없는 간선부터 모두 탐색한 후 가중치가 낮은 순서대로 탐색한다.

 

 

 

---

• 소스코드 : 

import sys
from heapq import heappop,heappush
input = sys.stdin.readline

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

N = int(input())
board = [list(map(int,input().strip())) for _ in range(N)]
visited = [[-1]*N for _ in range(N)]
heap = []
heappush(heap,[0,0,0])
visited[0][0] = 1
while heap:
    wall, x, y = heappop(heap)
    if x == N-1 and y == N-1:
       print(wall)
    for i in range(4):
        ax = x + dx[i]
        ay = y + dy[i]
        if 0 <= ax < N and 0 <= ay < N:
            if board[ax][ay] == 1 and visited[ax][ay] == -1:
                heappush(heap,[wall,ax,ay])
                visited[ax][ay] = 1
            if board[ax][ay] == 0 and visited[ax][ay] == -1:
                heappush(heap,[wall+1,ax,ay])
                visited[ax][ay] = 1