백준 알고리즘
[Silver II] 종이의 개수 - 1780
성능 요약
메모리: 48180 KB, 시간: 252 ms
분류
분할 정복, 재귀
문제 설명
N×N크기의 행렬로 표현되는 종이가 있다. 종이의 각 칸에는 -1, 0, 1 중 하나가 저장되어 있다. 우리는 이 행렬을 다음과 같은 규칙에 따라 적절한 크기로 자르려고 한다.
- 만약 종이가 모두 같은 수로 되어 있다면 이 종이를 그대로 사용한다.
- (1)이 아닌 경우에는 종이를 같은 크기의 종이 9개로 자르고, 각각의 잘린 종이에 대해서 (1)의 과정을 반복한다.
이와 같이 종이를 잘랐을 때, -1로만 채워진 종이의 개수, 0으로만 채워진 종이의 개수, 1로만 채워진 종이의 개수를 구해내는 프로그램을 작성하시오.
입력
첫째 줄에 N(1 ≤ N ≤ 37, N은 3k 꼴)이 주어진다. 다음 N개의 줄에는 N개의 정수로 행렬이 주어진다.
출력
첫째 줄에 -1로만 채워진 종이의 개수를, 둘째 줄에 0으로만 채워진 종이의 개수를, 셋째 줄에 1로만 채워진 종이의 개수를 출력한다.
Code
- 시간초과때매 애먹은 문제
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
use std::io;
fn paper(x: usize, y: usize, size: usize, map: &mut Vec<Vec<i32>>, one: &mut usize, m_one: &mut usize, ze: &mut usize) {
let mut same = true;
for i in y..y+size {
for j in x..x+size {
if map[i][j] != map[y][x] {
same = false;
break;
}
}
if !same {
break;
}
}
if !same {
paper(x, y, size / 3, map, one, m_one, ze);
paper(x + size / 3, y, size / 3, map, one, m_one, ze);
paper(x + size / 3 + size / 3, y, size / 3, map, one, m_one, ze);
paper(x, y + size / 3, size / 3, map, one, m_one, ze);
paper(x, y + size / 3 + size / 3, size / 3, map, one, m_one, ze);
paper(x + size / 3, y + size / 3, size / 3, map, one, m_one, ze);
paper(x + size / 3 + size / 3, y + size / 3 + size / 3, size / 3, map, one, m_one, ze);
paper(x + size / 3, y + size / 3 + size / 3, size / 3, map, one, m_one, ze);
paper(x + size / 3 + size / 3, y + size / 3, size / 3, map, one, m_one, ze);
} else {
match map[y][x] {
1 => *one += 1,
-1 => *m_one += 1,
_ => *ze += 1,
}
}
}
fn main() {
let mut input = String::new();
io::stdin().read_line(&mut input).expect("Failed to read line");
let n: usize = input.trim().parse().expect("Invalid input");
let mut map = vec![vec![0; n]; n];
for i in 0..n {
input.clear();
io::stdin().read_line(&mut input).expect("Failed to read line");
let row: Vec<i32> = input
.split_whitespace()
.map(|s| s.parse().expect("Invalid input"))
.collect();
map[i] = row;
}
let mut one = 0;
let mut m_one = 0;
let mut ze = 0;
paper(0, 0, n, &mut map, &mut one, &mut m_one, &mut ze);
println!("{}\n{}\n{}", m_one, ze, one);
}