💡문제접근
- BFS 탐색 알고리즘을 이용해서 문제를 해결할 수 있었다.
- 처음 제출한 코드는 2차원 배열에서 최솟값부터 최댓값까지
range
로 범위를 지정해서 탐색한 코드였는데 해당 코드가 WA를 받아서 문제를 자세히 읽어봤는데 문제 하단부에 노트
라는 부분에 힌트가 있었다.
- 아무 지역도 물에 잠기지 않을 수 있다. 비의 양을 나타내는 for문의 범위를 0부터 탐색해야한다.
💡코드(메모리 : 34192KB, 시간 : 1044ms)
from collections import deque
import sys
input = sys.stdin.readline
N = int(input().strip())
area = [list(map(int, input().strip().split())) for _ in range(N)]
def BFS(x, y, rain):
cnt = 0
queue = deque()
queue.append((x, y))
visited[x][y] = True
while queue:
x, y = queue.popleft()
dx = [0, 1, 0, -1]
dy = [-1, 0, 1, 0]
for i in range(4):
nx = x + dx[i]
ny = y + dy[i]
if nx < 0 or nx >= N or ny < 0 or ny >= N:
continue
if 0 <= nx < N and 0 <= ny < N and area[nx][ny] > rain and not visited[nx][ny]:
visited[nx][ny] = True
queue.append((nx, ny))
result = 0
for h in range(101):
cnt = 0
visited = [[False] * N for _ in range(N)]
for i in range(N):
for j in range(N):
if area[i][j] > h and not visited[i][j]:
BFS(i, j, h)
cnt += 1
if result < cnt:
result = cnt
print(result)
💡소요시간 : 10m