[BOJ][⚪1][백준#01926] 그림

문제 출처

BAEKJOON Online Judge #1926

문제

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

입력

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

출력

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

예제

입력

6 5
1 1 0 1 1
0 1 1 0 0
0 0 0 0 0
1 0 1 1 1
0 0 1 1 1
0 0 1 1 1

출력

4
9

My Sol

import sys
input = sys.stdin.readline
from collections import deque

def bfs(ni, nj):
    global maxArea, mat

    Q = deque()
    Q.append((ni, nj))
    cnt = 0
    while Q:
        ni, nj = Q.popleft()
        if not mat[ni][nj]: continue
        mat[ni][nj] = 0
        cnt += 1
        for di, dj in ((-1,0),(1,0),(0,-1),(0,1)):
            si, sj = ni+di, nj+dj
            if not (0<=si<I and 0<=sj<J): continue
            if not mat[si][sj]: continue
            Q.append((si, sj))

    maxArea = max(maxArea, cnt)


I, J = map(int, input().split())
mat = [list(map(int, input().split())) for _ in range(I)]
total = 0
maxArea = 0
for i in range(I):
    for j in range(J):
        if mat[i][j]:
            total += 1
            bfs(i, j)

print(total)
print(maxArea)

전형적인 BFS 문제였다. 모든 좌표를 완전탐색하면서, 1인 지점을 발견하면 BFS를 돌려 상하좌우로 연결된 영역을 모두 0으로 처리하고 전체 total 카운트를 1 추가해준다. 이때 BFS를 돌리며 영역의 수를 카운트 해주고, BFS를 마치면 전역 변수 maxArea와 비교해서 더 큰 값으로 갱신해준다.

모든 좌표에 대한 완전탐색이 끝나고 total과 maxArea를 출력하면 되겠다.

결과

맞았습니다!!