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

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

문제

눈금의 간격이 1인 $M×N(M,N≤100)$크기의 모눈종이가 있다. 이 모눈종이 위에 눈금에 맞추어 K개의 직사각형을 그릴 때, 이들 $K$개의 직사각형의 내부를 제외한 나머지 부분이 몇 개의 분리된 영역으로 나누어진다.

예를 들어 $M=5, N=7$ 인 모눈종이 위에 <그림 1>과 같이 직사각형 3개를 그렸다면, 그 나머지 영역은 <그림 2>와 같이 3개의 분리된 영역으로 나누어지게 된다.

<그림 2>와 같이 분리된 세 영역의 넓이는 각각 1, 7, 13이 된다.

$M, N$과 $K$ 그리고 $K$개의 직사각형의 좌표가 주어질 때, $K$개의 직사각형 내부를 제외한 나머지 부분이 몇 개의 분리된 영역으로 나누어지는지, 그리고 분리된 각 영역의 넓이가 얼마인지를 구하여 이를 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 $M$과 $N$, 그리고 $K$가 빈칸을 사이에 두고 차례로 주어진다. $M, N, K$는 모두 100 이하의 자연수이다. 둘째 줄부터 $K$개의 줄에는 한 줄에 하나씩 직사각형의 왼쪽 아래 꼭짓점의 $x, y$ 좌표값과 오른쪽 위 꼭짓점의 $x, y$ 좌표값이 빈칸을 사이에 두고 차례로 주어진다. 모눈종이의 왼쪽 아래 꼭짓점의 좌표는 $(0,0)$ 이고, 오른쪽 위 꼭짓점의 좌표는 $(N,M)$이다. 입력되는 $K$개의 직사각형들이 모눈종이 전체를 채우는 경우는 없다.

출력

첫째 줄에 분리되어 나누어지는 영역의 개수를 출력한다. 둘째 줄에는 각 영역의 넓이를 오름차순으로 정렬하여 빈칸을 사이에 두고 출력한다.

---

문제 풀이

1. 각 영역의 범위를 1로 초기화 한다.
2. 아직 방문하지 않고, arr[i][j]가 1이 아닐 경우만 dfs에 넣는다.
3. dfs 안에서의 횟수가 영역의 넓이 cnt 이고, 방문하지 않고 arr[i][j]가 1이 아닌 경우의 횟수 ans 가 총 영역의 수이다.

import sys
input = sys.stdin.readline
sys.setrecursionlimit(10**6)

m, n, k = map(int, input().split())
arr = [[0] * n for _ in range(m)]

for _ in range(k):
    left_X, left_Y, right_X, right_Y = map(int, input().split())

    for i in range(left_Y, right_Y):
        for j in range(left_X, right_X):
            arr[i][j] = 1


def dfs(x, y):

    global cnt
    visited[x][y] = True
    dx, dy = [-1, 1, 0, 0], [0, 0, -1, 1]
    cnt += 1
    for i in range(4):
        nx = x + dx[i]
        ny = y + dy[i]

        if nx < 0 or nx >= m or ny < 0 or ny >= n:
            continue

        if arr[nx][ny] != 1 and not visited[nx][ny]:
            visited[nx][ny] = True
            dfs(nx, ny)


visited = [[False] * n for _ in range(m)]
cntlist = []
ans = 0
for i in range(m):
    for j in range(n):
        if not visited[i][j] and arr[i][j] != 1:
            cnt = 0
            dfs(i, j)
            ans += 1
            cntlist.append(cnt)

cntlist.sort()
print(ans)
for cnt in cntlist:
    print(cnt, end=' ')