Algorithm/BaekJoon
[백준/Python] 2573번 - 빙산
빵빵0
2023. 9. 6. 00:29
import sys
from collections import deque
input = sys.stdin.readline
n, m = map(int, input().split())
graph = [list(map(int, input().split())) for _ in range(n)]
def bfs(x, y):
q = deque([(x, y)])
visited[x][y] = 1
seaList = []
while q:
x, y = q.popleft()
sea = 0
for i in range(4):
nx = x + dx[i]
ny = y + dy[i]
if 0 <= nx < n and 0 <= ny < m:
if not graph[nx][ny]:
sea += 1
elif graph[nx][ny] and not visited[nx][ny]:
q.append((nx, ny))
visited[nx][ny] = 1
if sea > 0:
seaList.append((x, y, sea))
# 주변 바다를 모두 탐색한 후에 한꺼번에 녹이기
for x, y, sea in seaList:
graph[x][y] = max(0, graph[x][y] - sea)
return 1 # 하나의 그룹을 탐색함
# 빙산만 리스트에 선별
ice = []
for i in range(n):
for j in range(m):
if graph[i][j]:
ice.append((i, j))
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]
year = 0
while ice:
visited = [[0] * m for _ in range(n)] # 연도마다 visited 갱신
delList = [] # 바다가 된 빙산
group = 0
for i, j in ice:
# 빙산 그룹 체크, 1년 경과
if graph[i][j] and not visited[i][j]:
group += bfs(i, j)
# 탐색이 끝나면 바다가 된 빙산 체크
if graph[i][j] == 0:
delList.append((i, j))
# 빙산 그룹이 2개 이상이면 해당 연도 출력
if group > 1:
print(year)
break
# 다 녹은 빙산은 탐색할 필요가 없으므로 ice에서 제거
ice = sorted(list(set(ice) - set(delList)))
year += 1
# 빙산이 다 녹았으면 0 출력
if group < 2:
print(0)
옛날에 풀었던 문제다. 나중에 다시 풀어보고 해석을 남겨야겠다
# visited의 용도
## bfs 로직으로서 확인했던 좌표인지 확인
## 빙산 group을 확인!! -> 사실 이렇게 안하고 그냥 밑에 count += 1 이런거 해도 됨