[백준] 2583번 영역 구하기 - 파이썬

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

2583번: 영역 구하기

import sys
from collections import deque
m, n, k = map(int,sys.stdin.readline().split())
graph = [[0]*n for i in range(m)]

for _ in range(k):
    x1,y1,x2,y2 = map(int,sys.stdin.readline().split())

    for i in range(y1,y2):
        for j in range(x1,x2):
            graph[i][j] = 1

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

def bfs(x,y,k):
    q = deque()
    q.append((x,y))
    count = 0
    while q:
        x,y =q.popleft()
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            if 0 <= nx < m and 0 <= ny < n:
                if graph[nx][ny] == 0 :
                    graph[nx][ny] = 1
                    count += 1
                    q.append((nx, ny))
    k += 1
    if count == 0 :
        count = 1
    return count, k

part =0
rec_cnt = 0
part_list = []

for i in range(len(graph)):
    for j in range(len(graph[i])):
        if graph[i][j] == 0:
            rec_cnt, part=bfs(i,j,part)
            part_list.append(rec_cnt)

print(part)
part_list.sort()
for i in range(len(part_list)):
    print(part_list[i], end = " ")

어렵지는 않은 문제이다.

알고리즘을 잘 이해하고 있다면 쉽게 풀 수 있다. 하지만 

문제에 주어진 입력을 잘 받는 방법만 조금 생각하면된다.