[Swift Algorithm] 영역 구하기 BOJ #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개의 직사각형들이 모눈종이 전체를 채우는 경우는 없다.

 

출력

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

 

입출력 예제

입력

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

출력

3
1 7 13

 

내 문제 풀이

let info = readLine()!.split(separator: " ").map({Int(String($0))!})
let m = info[0]
let n = info[1]
let k = info[2]
var paper = Array(repeating: Array(repeating: 0, count: n), count: m)
var area = [Int]()
var area_tmp = 0

// 직사각형들 그려주기
for _ in 0..<k {
    let draw = readLine()!.split(separator: " ").map({Int(String($0))!}) // 직사각형 좌표
    
    for i in draw[0]..<draw[2] {
        for j in draw[1]..<draw[3] {
            paper[j][i] += 1
        }
    }
}

// dfs 재귀함수
func dfs(_ x: Int, _ y: Int) {
    if x < 0 || x >= m || y < 0 || y >= n || paper[x][y] != 0 { return }
    
    paper[x][y] = 1 // 이미 체크한 자리는 1로 바꿔주기
    area_tmp += 1 // area_tmp 1 증가시켜주기
    
    // 재귀 호출 (현재 자리와 붙어있는 주변 모든 자리 확인)
    dfs(x+1, y) // 세로로 다음 자리
    dfs(x-1, y) // 세로로 전 자리
    dfs(x, y+1) // 가로로 다음 자리
    dfs(x, y-1) // 가로로 전 자리
}

// 메인 실행문
for i in 0..<n {
    for j in 0..<m {
        if paper[j][i] == 0 {
            area_tmp = 0
            dfs(j, i)
            area.append(area_tmp)
        }
    }
}

print(area.count)
print(area.sorted().map({String($0)}).joined(separator: " "))

👉 dfs 문제로, 먼저 직사각형 자리들을 1로 채워준 후 0으로 이어진 영역들을 찾는게 포인트!

(이전에 게시했던 #1926 그림 문제와 아주 유사하므로 간단히 설명하겠다.)

• 가장 먼저 k 값을 이용하여 직사각형 자리들을 채워주었다.
  for 반복을 이용했고, 직사각형이 그려질 자리에는 1을 더했다.
  => 그림이 그려진 자리는 1 이상의 수가 채워지고, 그 외의 자리는 모두 0이 됨
• 2중 for문을 이용해서 종이(paper 배열)을 탐색하다가, 0을 만나면 dfs를 실행했다.
  dfs 함수는 재귀로 구현했고, 체크 완료한 자리는 1로 바꿔주면서 넓이를 구했다.

 

💡 피드백

• 이전에 풀었던 문제와 유사해서 금방 풀 수 있었다.
• 좌표가 좌측 하단부터 우측 하단으로 뻗어나가는 것으로 주어졌기 때문에
  배열의 좌표를 지정해주는 부분에서 조금 헷갈렸는데, 좌표만 반대로 생각하면 쉽게 해결할 수 있었다.

 

---

문제

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