문제
n×n 바둑판 모양으로 총 n2개의 방이 있다. 일부분은 검은 방이고 나머지는 모두 흰 방이다. 검은 방은 사면이 벽으로 싸여 있어 들어갈 수 없다. 서로 붙어 있는 두 개의 흰 방 사이에는 문이 있어서 지나다닐 수 있다. 윗줄 맨 왼쪽 방은 시작방으로서 항상 흰 방이고, 아랫줄 맨 오른쪽 방은 끝방으로서 역시 흰 방이다.
시작방에서 출발하여 길을 찾아서 끝방으로 가는 것이 목적인데, 아래 그림의 경우에는 시작방에서 끝 방으로 갈 수가 없다. 부득이 검은 방 몇 개를 흰 방으로 바꾸어야 하는데 되도록 적은 수의 방의 색을 바꾸고 싶다.
아래 그림은 n=8인 경우의 한 예이다.
위 그림에서는 두 개의 검은 방(예를 들어 (4,4)의 방과 (7,8)의 방)을 흰 방으로 바꾸면, 시작방에서 끝방으로 갈 수 있지만, 어느 검은 방 하나만을 흰 방으로 바꾸어서는 불가능하다. 검은 방에서 흰 방으로 바꾸어야 할 최소의 수를 구하는 프로그램을 작성하시오.
단, 검은 방을 하나도 흰방으로 바꾸지 않아도 되는 경우는 0이 답이다.
입력
첫 줄에는 한 줄에 들어가는 방의 수 n(1≤n≤50)이 주어지고, 다음 n개의 줄의 각 줄마다 0과 1이 이루어진 길이가 n인 수열이 주어진다. 0은 검은 방, 1은 흰 방을 나타낸다.
출력
첫 줄에 흰 방으로 바꾸어야 할 최소의 검은 방의 수를 출력한다.
예제 입력과 출력
정답
from collections import deque
n=int(input())
a=[list(map(int,input())) for i in range(n)]
ch=[[-1] * n for i in range(n)]
dx=[-1,0,1,0]
dy=[0,1,0,-1]
def bfs():
de=deque()
de.append([0,0])
ch[0][0]=0
while de:
x,y=de.popleft()
for i in range(4):
nx = x + dx[i]
ny = y + dy[i]
if 0 <= nx < n and 0 <= ny < n:
if ch[nx][ny] == -1:
if a[nx][ny] == 0:
ch[nx][ny]= ch[x][y] + 1
de.append([nx,ny])
else:
ch[nx][ny]= ch[x][y]
de.appendleft([nx,ny])
bfs()
print(ch[n-1][n-1])
흰방에 도착하면 이동거리는 이전과 같고 먼저 흰방을 방문하기 위해 appendleft를 사용합니다.
검은방에 도착하면 이동거리는 이전에서 +1해주고 append를 사용합니다.
백준 1261 알고스팟과 똑같은 문제입니다.
백준 알고리즘 2665번 : https://www.acmicpc.net/problem/2665
'알고리즘 > 백준알고리즘' 카테고리의 다른 글
백준알고리즘 - 3986번 좋은 단어 - 파이썬(Python) (0) | 2020.06.18 |
---|---|
백준알고리즘 - 1874번 스택 수열 - 파이썬(Python) (0) | 2020.06.18 |
백준알고리즘 - 2638번 치즈 - 파이썬(Python) (0) | 2020.06.17 |
백준알고리즘 - 2636번 치즈 - 파이썬(Python) (0) | 2020.06.17 |
백준알고리즘 - 1987번 알파벳 - 파이썬(Python) (0) | 2020.06.17 |