[백준] 1926번 그림 파이썬

문제

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

입력

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

출력

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

예제 입력 1

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

예제 출력 1

4
9

제출답안

from collections import deque
import sys

input = sys.stdin.readline


def bfs(n, m, x, y, graph, dx, dy, visited):
    distance = 0
    queue = deque()
    queue.append((x, y, 0))
    visited[x][y] = True
    while queue:
        x, y, d = 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 and visited[nx][ny] == False:
                distance += 1
                queue.append((nx, ny, distance))
                visited[nx][ny] = True
    return distance, visited


def solution():
    dx = [0, 0, -1, 1]
    dy = [-1, 1, 0, 0]
    area_list = []
    n, m = map(int, input().split())
    # n, m = 6, 5
    graph = [list(map(int, input().split())) for _ in range(n)]
    # graph = [[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]]
    visited = [[False] * m for _ in range(n)]
    for i in range(n):
        for j in range(m):
            if graph[i][j] == 1 and visited[i][j] == False:
                distance, visited = bfs(n, m, i, j, graph, dx, dy, visited)
                area_list.append(distance)
    print(len(area_list))
    if len(area_list) == 0:
        print(0)
    else:
        print(max(area_list) + 1)


solution()

1. bfs 함수를 만든다.
   1. 덱을 사용하여 선입선출을 사용한다.
   2. 먼저 start를 삽입하고 visited도 True로 바꿔준다.
   3. while을 돌면서 x,y,거리값을 먼저 들어온 것 부터 뽑는다.
   4. 해당 x,y의 상하좌우를 보면서 1이면서 방문하지 않았으면 거리에 1을 추가하고 해당 좌표를 True로 변경하고 덱에도 값을 넣는다.
2. 이렇게 n,m for 문을 돌면서 bfs를 실행하면 거리를 가져올 수 있다.
3. 리스트에 담긴 갯수와 최댓값을 print 한다.