[백준] 1926 - 그림(파이썬)

문제

어떤 큰 도화지에 그림이 그려져 있을 때, 그 그림의 개수와, 그 그림 중 넓이가 가장 넓은 것의 넓이를 출력하여라. 단, 그림이라는 것은 1로 연결된 것을 한 그림이라고 정의하자. 가로나 세로로 연결된 것은 연결이 된 것이고 대각선으로 연결이 된 것은 떨어진 그림이다. 그림의 넓이란 그림에 포함된 1의 개수이다.

입력

첫째 줄에 도화지의 세로 크기 n(1 ≤ n ≤ 500)과 가로 크기 m(1 ≤ m ≤ 500)이 차례로 주어진다. 두 번째 줄부터 n+1 줄 까지 그림의 정보가 주어진다. (단 그림의 정보는 0과 1이 공백을 두고 주어지며, 0은 색칠이 안된 부분, 1은 색칠이 된 부분을 의미한다)

출력

첫째 줄에는 그림의 개수, 둘째 줄에는 그 중 가장 넓은 그림의 넓이를 출력하여라. 단, 그림이 하나도 없는 경우에는 가장 넓은 그림의 넓이는 0이다.

---

풀이

bfs 활용

from collections import deque

n, m = map(int, input().split())
grid = []

# 초기화
for _ in range(n):
    grid.append(list(map(int, input().split())))

# 한 구역의 넓이를 구하는 함수
def bfs_width(start, grid):
    queue = deque([start])
    grid[start[0]][start[1]] = 0
    width = 0

    directions = [(-1, 0), (1, 0), (0, 1), (0, -1)]

    while queue:
        position = queue.popleft()
        width += 1
        
        for da, db in directions:
            a = position[0] + da
            b = position[1] + db

            if 0 <= a < n and 0 <= b < m:
                if grid[a][b] == 1:
                    queue.append((a, b))
                    grid[a][b] = 0

    return width

# 답을 구하는 과정
count = 0
max_width = 0

for i in range(n):
    for j in range(m):
        if grid[i][j] == 1:
            count += 1
            width = bfs_width((i, j), grid)
            max_width = max(width, max_width)

print(count)
print(max_width)