Baekjoon [1940] 그림 - python

그림 성공

문제

어떤 큰 도화지에 그림이 그려져 있을 때, 그 그림의 개수와, 그 그림 중 넓이가 가장 넓은 것의 넓이를 출력하여라. 단, 그림이라는 것은 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
n,m = map(int,input().split())

mapp = []
for i in range(n):
    mapp.append(list(map(int,input().split())))
visited = [[0 for _ in range(m)] for _ in range(n)]
cnt = 0
def bfs(i,j):
    que = deque()
    check_num = 0
    dx = [1,-1,0,0]
    dy = [0,0,-1,1]
    que.append((j,i))
    while que:
        x,y = que.popleft()
        if visited[y][x] == 1:
            pass
        else:
            check_num +=1
            visited[y][x] = 1
            for i in range(4):
                xx = dx[i] + x
                yy = dy[i] + y
                if 0<= xx < m and 0<= yy < n:
                    if mapp[yy][xx] == 1 and visited[yy][xx] == 0:
                        que.append((xx,yy))
    return check_num

answer_list = []
for i in range(n):
    for j in range(m):
        if visited[i][j] == 0 and mapp[i][j] == 1:
            a = bfs(i,j) # y,x
            cnt +=1
            answer_list.append(a)

if len(answer_list) == 0:
    print(cnt)
    print(0)
else:
    print(cnt)
    print(max(answer_list))