[백준 알고리즘] 1926번 그림, 파이썬(python)

1926, 그림

📁 문제 출처

1926번: 그림

 

💡 생각

• 그림의 개수와, 그 그림 중 넓이가 가장 넓은 것의 넓이를 출력하는 문제
• 모든 부분을 탐색해야 되기 때문에 BFS로 풀이
• 이어져 있는 부분은 count 변수로 카운트해주고 paint 배열에 append
• 그림의 개수는 len(paint), 가장 넓은 그림은 max(paint)

 

🛠 나의 코드

from collections import deque

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

for i in range(n):
    graph.append(list(map(int, input().split())))

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

def bfs(graph, a, b):
    queue = deque()
    queue.append((a, b))
    graph[a][b] = 0
    count = 1

    while queue:
        x, y = queue.popleft()
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            if nx < 0 or nx >= n or ny < 0 or ny >= m:
                continue
            if graph[nx][ny] == 1:
                graph[nx][ny] = 0
                queue.append((nx, ny))
                count += 1
    return count

paint = []
for i in range(n):
    for j in range(m):
        if graph[i][j] == 1:
            paint.append(bfs(graph, i, j))


if len(paint) == 0:
    print(len(paint))
    print(0)
else:
    print(len(paint))
    print(max(paint))

 

✔️ 배운 점 및 주의할 점

• 오랜만에 BFS를 풀어서 이전에 정리한 기본 개념을 훑어보고 풀었다.
• BFS의 가장 기본적인 문제였다.
• DFS 문제 몰아 풀 때 다시 풀어볼 것