[백준/BOJ][Python] 2583번 영역 구하기

[백준/BOJ][Python] 2583번 영역 구하기

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

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 

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

예제 출력 1 

3
1 7 13

---

생각한 과정

직사각형이 있는 곳에는 graph 값을 1로 만들고

graph값이 0인 곳만 BFS를 돌면서 넓이를 세줘야한다는 생각이 문제를 보자마자 들었다

 

직사각형이 있는 곳의 graph 값을 1로 만드는 건 *옛날에 풀었던 문제랑 비슷해서 풀이 방법이 떠오른 거고 

안 풀었으면 오래 걸렸을 것 같은 문제이다.

 

* 코드업 1507번 4개의 직사각형 넓이

https://dkan9634.tistory.com/138

[CodeUp][Python] 1507번 4개의 직사각형 넓이

 

 

우선 첫 번째 줄에서는 sys와 collections 모듈을 가져오고, input() 함수 대신 sys.stdin.readline() 함수를 사용했다.

이는 입력 속도를 더 빠르게 만들어준다. 그 다음으로는 bfs 함수를 정의

 

bfs 함수는 시작점 a와 b를 받고, 이후에는 deque를 사용하여 큐를 구현하고, 시작점의 좌표를 큐에 추가

이후에는 area 변수를 1로 초기화하고, 큐가 빌 때까지 반복

 

큐에서 좌표를 하나씩 꺼내어 상하좌우로 이동

그리고 만약 이동한 좌표가 그래프 범위 안에 있고, 그래프에서 이동한 좌표에 해당하는 값이 0이라면(즉, 직사각형이 아니라면) area 값을 1 증가시키고, 그래프에서 이동한 좌표에 1을 대입하고, 이동한 좌표를 큐에 추가

 

반복문이 끝난 후에는 area 값이 1이 아니라면 area 값을 1 감소시킵니다. 그리고 area 값을 return

 

다음으로는 M, N, K를 입력

이때 M은 그래프의 행 수, N은 그래프의 열 수, K는 직사각형의 개수를 의미

 

graph 배열은 좌표를 나타내고 초기에는 모든 값이 0

이후에는 K번 반복문을 실행하여, 직사각형이 위치한 영역에 대해 이중 반복문을 실행하면서 graph 배열에서 해당하는 위치를 1로 바꿔줌

마지막으로, 이중 반복문을 실행하여 graph 배열에서 값이 0인 좌표를 찾아 bfs 함수를 호출하여 직사각형의 개수를 찾는다.

이후에는 직사각형의 개수를 출력하고, 각 직사각형의 넓이 오름차순으로 출력

 

---

코드

import sys
from collections import deque
input = sys.stdin.readline

# BFS 함수
def bfs(a, b):
    queue = deque()
    queue.append([a, b])
    area = 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:
            	# 그래프 값이 0이 아니면 continue(값이 1이면 직사각형임)
                if graph[nx][ny] != 0:
                    continue
                    
                # 그래프 값이 0이면
                else:
                    area += 1 # 넓이 1 증가
                    graph[nx][ny] = 1 # graph 값 바꿔주고
                    queue.append([nx, ny]) # BFS 돌기 위해 큐에 넣어주기

    if area != 1:
        area -= 1
    return area


M, N, K = map(int, input().split())
graph = []
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]

for i in range(M):
    graph.append([0]*N)

# 전체적인 그래프 만들기
# 직사각형이 있는 곳에는 1, 없는 곳에는 0
# 따라서 0이 있는 곳만 BFS 돌면 된다
for _ in range(K):
    x1, y1, x2, y2 = map(int, input().split())
   
   # ********이부분이 생각하기 어려웠음
    for j in range(x1, x2):
        for k in range(y1, y2):
            graph[k][j] = 1

answer = []
for i in range(M):
    for j in range(N):
        if graph[i][j] == 0:
            answer.append(bfs(i, j))

print(len(answer))
answer.sort()
for i in range(len(answer)):
    print(answer[i], end=' ')