[BOJ | Python] 1926번: 그림

https://www.acmicpc.net/problem/1926

 

 

문제

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

 

입력

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

 

 

출력

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

 

 

풀이

from collections import deque

def BFS(x, y, area):
    dxs, dys = [1, -1, 0, 0], [0, 0, 1, -1]

    queue = deque([(x, y)])
    visited[x][y] = True

    while queue:
        x, y = queue.popleft()
        for dx, dy in zip(dxs, dys):
            if 0 <= x + dx < n and 0 <= y + dy < m:
                if not visited[x + dx][y + dy] and graph[x + dx][y + dy]:
                    visited[x + dx][y + dy] = True
                    area += 1
                    queue.append((x + dx, y + dy))
    return area 

n, m = map(int, input().split())
graph = [list(map(int, input().split())) for _ in range(n)]

ans = []
visited = [[False] * m for _ in range(n)]

for i in range(n):
    for j in range(m):
        if not visited[i][j] and graph[i][j]:
            ans.append(BFS(i, j, 1))

if not ans: print("0\n0")
else: print(f"{len(ans)}\n{max(ans)}")

 

런타임 에러가 발생했다면 이 부분을 주의하자..