[BOJ] 1926 도화지 (Python3)

문제

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

문제풀이

간단한 dfs 문제이다. n*m의 좌표의 요소에 하나씩 접근하여 그 값이 1이라면 그림의 개수를 1 더해주고, dfs를 실행하여 상하좌우로 이어진 1의 개수가 몇개인지 세주면된다.

코드

from collections import deque


dx = [1,-1,0,0]
dy = [0,0,1,-1]

#dfs 함수, pop할때마다 count+=1하여 최종 개수를 return한다.
def dfs(x, y):
    q = deque()
    if graph[x][y] == 1:
        q.append([x,y])
        visited[x][y] = 1
    count = 0
    while q:
        nx,ny = q.popleft()
        count+=1
        for i in range(4):
            sx = nx+dx[i]
            sy = ny+dy[i]
    
            if 0<= sx < n and 0<= sy < m and graph[sx][sy] == 1 and visited[sx][sy] == 0 :
                q.append([sx,sy])
                visited[sx][sy] = 1
                
                   
    return count         
        

n,m = map(int, input().split())

graph =[]

visited = [[0 for i in range(m)] for i in range(n)]

for i in range(n):
    graph.append(list(map(int,input().split(' '))))

width = []
num = 0

# graph의 모든 좌표를 접근하다가 1이 나오면 num+=1을 해주고 dfs를 실행한다.
for i in range(len(graph)):
    for j in range(len(graph[0])):
        if graph[i][j]== 1 and visited[i][j]== 0:
            width.append(dfs(i,j))
            num+=1

print(num)
if width:
    print(max(width))
else:
    print(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