[백준] 1926 : 그림 - Python

그림

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

입력

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

import sys
sys.stdin = open("input.text", "rt")
input = sys.stdin.readline
from collections import deque

#그림 개수와 그 그림 중 넓이가 가장 넓은 것의 넓이
#상하좌우
n, m = map(int, input().split())
g = [list(map(int,input().split())) for _ in range(n)]

#그림의 개수 = 연결요소
#그림의 넓이 = 1개수

dx = [1,-1,0,0]
dy = [0,0,1,-1]
cnt = 0
max_cnt = 0
def BFS(a,b):
    dq = deque()
    dq.append((a,b))
    g[a][b] = 0 #방문 처리
    res = 1 #시작점도 포함해야지.
    while dq:
        x,y = dq.popleft()
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            if 0<=nx<n and 0<=ny<m: 
                if g[nx][ny] == 1:
                    g[nx][ny] = 0 #방문 처리
                    res += 1
                    dq.append((nx,ny))
    return res

res = [0]
for i in range(n):
    for j in range(m):
        if g[i][j] == 1:
            temp = BFS(i,j)
            cnt += 1
            max_cnt = max(max_cnt, temp)

print(cnt)
print(max_cnt)

코멘트

해당 문제는 그래프 탐색 문제 중 DFS, BFS 모두를 활용하여 풀 수 있는 문제이다. 문제 내용을 이해해보면 그림의 개수 = 연결 요소의 개수이고, 그림 중 가장 넓은 것의 넓이는 가지고 있는 연결 요소의 개수 중에서 가장 많은 개수를 가진 것을 출력하면 된다 !

• 나는 해당 문제를 BFS로 해결했다.

👻 출력부분에서 그림이 없는 경우 넓이는 0 이라는 점에서 처음에는 res = [] 리스트를 만들어서 BFS 탐색이 끝날때마다 넣어줬는데 테스트 케이스에 답이 0인 것도 있었는지 런타임 에러가 떴다. 그렇기에 따로 받아서 크기 비교를 하는 방식으로 바꿔줬다.