백준 2583번 영역구하기

DFS 문제

문제

눈금의 간격이 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

5 7 3
0 2 4 4
1 1 2 5
4 0 6 2

예제 출력

3
1 7 13

답안

import sys
sys.setrecursionlimit(10 ** 7)
 
def dfs(x, y):
    global cnt
    if x <= -1 or x >= m or y <= -1 or y >= n:
        return False
    if graph[x][y] == 0:
        cnt +=1
        graph[x][y] = 1
        dfs(x - 1, y)
        dfs(x, y - 1)
        dfs(x + 1, y)
        dfs(x, y + 1)
        return True
    return False
 
 
m,n,k = map(int, input().split())
graph = [[0] * n for _ in range(m)]
 
for i in range(k):
    x1, y1, x2, y2 = map(int, input().split())
    for one in graph[y1:y2]:
        one[x1:x2] = map(lambda x: 1, one[x1:x2])
 
result = 0
answer = []
 
for i in range(m):
    for j in range(n):
        cnt = 0
        a = dfs(i, j)
        if a:
            answer.append(cnt)
            result += 1
 
print(result)
print(' '.join(map(str,sorted(answer))))

 

1. 우선 함수를 만든다. 범위를 벗어나거나 0이 아니면 False를 리턴시키고 0이면 1로 바꾸고 다른 dfs도 실행하도록 한다. 동시에 cnt의 숫자를 증가시킨다.
2. input으로 필요한 값들을 받는다. 그다음에 그래프를 m만큼 만들고 n만큼 다 0으로 초기화 시킨다.

0011000
1100111

3. 그러면 이렇게 바뀐다.
4. 이 리스트를 기준으로 for문을 돌려서 dfs를 실행시키고 True일 때 answer에 cnt를 저장하고 result를 증가시킨다.
5. 최종으로 결과와 리스트에 담긴 넓이들을 오름차순으로 정렬하여 출력한다.