[백준 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개의 직사각형들이 모눈종이 전체를 채우는 경우는 없다.

출력

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

풀이

정사각형이 위치하지 않은 영역의 각각의 크기를 구해야 한다.

인접한 영역중 빈 곳이 있으면 빈 곳 하나당 count를 1씩 증가해야 하므로

dfs를 이용하여 문제를 해결하였다.

우선 모눈종이를 표시할 matrix 리스트를 생성하고 0으로 초기화 하였다.

그리고 입력받은 사각형 위치의 눈금을 -1로 변경하였다.

이제 matrix 리스트를 순회하여 빈 눈금이 있고, 방문하지 않은 인덱스일 때 dfs 함수를 호출한다.

dfs 함수에서는 우선 해당 인덱스를 방문표시하고 count를 1 증가시킨다. 그리고 dx, dy 리스트를 이용하여

상하좌우 인접한 인덱스가 방문하지 않았고, 빈 공간이라면 dfs 함수를 재귀호출한다.

이렇게 재귀함수를 이용해 dfs를 구현하면 빈 공간의 개수를 셀 수 있다.

재귀함수가 끝나면 해당 count 값을 ans 리스트에 추가시킨다.

ans 리스트의 길이와 각 원소를 출력한다.

 

import sys
sys.setrecursionlimit(10000)
w, h, k = map(int, sys.stdin.readline().rstrip("\n").split())
matrix=[[0]*w for _ in range(h)]
ans = []
dx=[-1,1,0,0]
dy=[0,0,-1,1]
count=0
def dfs(q,p):
    global count
    matrix[q][p]=1
    count+=1
    for t in range(4):
        ny = q+dy[t]
        nx = p+dx[t]
        if(0<=ny<h and 0<=nx<w and matrix[ny][nx]==0):
            dfs(ny,nx)

for i in range(k):
    y1,x1,y2,x2=map(int, sys.stdin.readline().rstrip("\n").split())
    for q in range(y1, y2):
        for p in range(x1,x2):
            matrix[q][p]=-1


for q in range(h):
    for p in range(w):
        if(matrix[q][p]==0):
            count=0
            dfs(q,p)
            ans.append(count)

ans = sorted(ans)
print(len(ans))
for ansIndex in ans:
    print(ansIndex,end=" ")