2583 영역 구하기 bfs

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

2583번: 영역 구하기

from collections import deque

m,n,k = map(int,input().split())
#그래프 표기
graph = [[0]*n for _ in range(m)]
#그래프에 칸 채우기
for _ in range(k):
    x1,y1,x2,y2 = map(int,input().split())
    for i in range(x1,x2):
        #매트릭스에 맞게 좌표 다시 재설정
        for j in range(m-y1-1,m-y2-1,-1):
            graph[j][i] = 1

#하 상 좌 우            
dx=[-1,1,0,0]
dy=[0,0,-1,1]

def bfs(x,y):
    queue = deque()
    queue.append((x,y))
    graph[x][y] = 1
    blank = 1
    
    while queue:
        x,y = queue.popleft()
        for i in range(4):
            nx=x+dx[i]
            ny=y+dy[i]
            #범위 안에 있고 아직 안갔으면 
            if 0 <= nx <m and 0<= ny<n  and graph[nx][ny]==0:
                graph[nx][ny] =1
                queue.append((nx,ny))
                blank+=1
    result.append(blank)
    
result = []

for i in range(m):
    for j in range(n):
        if graph[i][j]==0:
            bfs(i,j)

#오름차순 정렬
result.sort()
print(len(result))
for i in result:
    print(i,end=' ')