[Baekjoon] 1926번 그림

문제

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

입력

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

출력

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

예제

Example 1:

Input: 
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
Output: 
4
9

조건

시간 제한 : 2초

메모리 제한 : 128 MB

풀이과정

내 풀이

간단하게 BFS를 이용하면 풀 수 있는 문제이다.

from collections import deque

# 세로, 가로 크기 입력
n, m = map(int ,input().split())

# 그래프 입력
graph= []
for _ in range(n):
    graph.append(list(map(int, input().split())))

# 이동 설정
dx = [-1, 0, 1, 0]
dy = [0, 1, 0, -1]
ans1 = 0
ans2 = 0
# BFS 정의
def bfs(x, y):
    global ans1
    global ans2
    sub = 0
    q = deque()
    if graph[x][y] == 1:
        graph[x][y] = 0
        sub += 1
        q.append((x, y))
        ans1 += 1

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


for x in range(n):
    for y in range(m):
        bfs(x, y)

print(ans1)
print(ans2)