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

백준 2583_영역 구하기

시간 제한 1초, 메모리 제한 128MB

# 조건

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

출력

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

# 접근 방법

• 우선 모눈종이의 왼쪽 아래 꼭지점이 (0, 0)이고, 오른쪽 위 꼭지점이 (N, M)인 것을 파악해야 한다..
• 또한 전체 모눈종이를 상하반전해도 문제가 없으므로 주어진 좌표를 왼쪽 위 ~ 오른쪽 아래로 사용해준다.
  • 주의할 점은 열, 행 순서로 주어지므로 잘 바꿔서 사용하자.
  • 또한 각 칸의 좌표가 아닌 모서리를 나타내기 때문에 오른쪽 아래의 좌표 (주어지는 오른쪽 위의 좌표) 까지를 범위의 마지막으로 사용해주면 된다.
• 이후에는 BFS를 사용하여 나누어지는 영역의 넓이를 RESULT 배열에 저장해주고 길이와 오름 차순 정렬 후 출력해주면 된다.

# python 풀이

import sys  
sys.stdin = open('input.txt')  
input = sys.stdin.readline  
from collections import deque  

def bfs(bi, bj):  
    q = deque()  
    q.append((bi, bj))  
    cnt = 1  
    while q:  
        si, sj = q.popleft()  
        for d in range(4):  
            ni, nj = si+di[d], sj+dj[d]  
            if 0<=ni<N and 0<=nj<M and arr[ni][nj] == 0 and not visited[ni][nj]:  
                visited[ni][nj] = 1  
                cnt += 1  
                q.append((ni, nj))  

    return cnt  


N, M, K = map(int, input().split())  
visited = [[0] * M for _ in range(N)]  
arr = [[0] * M for _ in range(N)]  
di, dj = [1, -1, 0, 0], [0, 0, 1, -1]  
result = []  
for _ in range(K):  
    b, a, d, c = map(int, input().split())  
    for i in range(a, c):  
        for j in range(b, d):  
            arr[i][j] = 1  

for i in range(N):  
    for j in range(M):  
        if arr[i][j] == 0 and not visited[i][j]:  
            visited[i][j] = 1  
            area = bfs(i, j)  
            result.append(area)  
print(len(result))  
result.sort()  
print(*result)

cpp 풀이

#include <iostream>
#include <queue>
#include <algorithm>

using namespace std;

int arr[100][100];
int N, M, K;
int di[4] = { 1, 0, -1, 0 }, dj[4] = { 0, 1, 0, -1 };
queue<pair<int, int>> q;

int bfs() {
    int cnt = 0; //빈 칸이 하나밖에 없는 영역일 경우 그대로 빠져나와버림
    while (!q.empty()) {
        int x = q.front().second;
        int y = q.front().first;
        q.pop();
        for (int i = 0; i < 4; i++) {
            int nx = x + di[i];
            int ny = y + dj[i];
            if (nx < 0 || ny < 0 || nx >= N || ny >= M)continue;
            if (!arr[ny][nx]) {
                cnt++;
                arr[ny][nx] = 1;
                q.push({ ny,nx });
            }
        }
    }
    return cnt;
}

int main() {
    int x1, x2, y1, y2, cnt = 0, area[100];
    cin >> M >> N >> K;
    while (K--) {
        cin >> x1 >> y1 >> x2 >> y2;
        for (int i = y1; i < y2; i++)
            for (int j = x1; j < x2; j++)
                arr[i][j] = 1;
    }
    for (int i = 0; i < M; i++) {
        for (int j = 0; j < N; j++) {
            if (!arr[i][j]) {
                q.push({ i, j });
                area[cnt++] = bfs();
                if (!area[cnt - 1])
                    area[cnt - 1] = 1;
            }
        }
    }
    sort(area, area + cnt);
    cout << cnt << "\n";
    for (int i = 0; i < cnt; i++)cout << area[i] << " ";
    return 0;
}