문제
풀이
저번에 풀었던 분할정복 문제와 유사하다. 백준 1074번 Z 풀이
재귀 함수를 이용하여 분할정복을 구현했다.
재귀 함수의 인자인 y,x는 현재 탐색하고자 하는 사분면의 가장 왼쪽 위의 좌표이며, size는 사분면의 한 변의 길이이다.
1
2
void solve(int y, int x, int size)
{
check 변수에는 현재 y,x 좌표에 있는 색종이의 색을 할당해주고 이중 for문을 통해 check와 다른 색상이 현재 사분면에 존재하는지 확인한다.
1
2
3
void solve(int y, int x, int size)
{
int check = paper[y][x];
만약 check와 다른 색상이 존재한다면 자를 수 없으므로 현재 사분면을 4등분해서 각각의 나눠진 4분면을 재귀 함수로 호출해주고 return해준다.
현재 사분면이 모두 같은 색이라면 check의 색상에 따라 blue 또는 white 변수에 카운트해준다.
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
void solve(int y, int x, int size)
{
int check = paper[y][x];
for (int i = y; i < y + size; i++)
{
for (int j = x; j < x + size; j++)
{
if (check == 0 && paper[i][j] == 1)
{
check = 2;
}
else if (check == 1 && paper[i][j] == 0)
{
check = 2;
}
if (check == 2)
{
// 왼쪽 위 사분면 탐색
solve(y, x, size / 2);
// 오른쪽 위 사분면 탐색
solve(y, x + size / 2, size / 2);
// 왼쪽 아래 사분면 탐색
solve(y + size / 2, x, size / 2);
// 오른쪽 아래 사분면 탐색
solve(y + size / 2, x + size / 2, size / 2);
return;
}
}
}
if (check == 0)
white++;
else
blue++;
}
소스 코드
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
#include <iostream>
using namespace std;
int n;
int paper[128][128];
int blue, white;
void solve(int y, int x, int size)
{
int check = paper[y][x];
for (int i = y; i < y + size; i++)
{
for (int j = x; j < x + size; j++)
{
if (check == 0 && paper[i][j] == 1)
{
check = 2;
}
else if (check == 1 && paper[i][j] == 0)
{
check = 2;
}
if (check == 2)
{
solve(y, x, size / 2);
solve(y, x + size / 2, size / 2);
solve(y + size / 2, x, size / 2);
solve(y + size / 2, x + size / 2, size / 2);
return;
}
}
}
if (check == 0)
white++;
else
blue++;
}
int main()
{
ios_base::sync_with_stdio(false);
cin.tie(NULL);
cout.tie(NULL);
cin >> n;
for (int i = 0; i < n; i++)
for (int j = 0; j < n; j++)
cin >> paper[i][j];
solve(0, 0, n);
cout << white << '\n';
cout << blue << '\n';
return 0;
}
