728x90

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

 

2630번: 색종이 만들기

첫째 줄에는 전체 종이의 한 변의 길이 N이 주어져 있다. N은 2, 4, 8, 16, 32, 64, 128 중 하나이다. 색종이의 각 가로줄의 정사각형칸들의 색이 윗줄부터 차례로 둘째 줄부터 마지막 줄까지 주어진다. 하얀색으로 칠해진 칸은 0, 파란색으로 칠해진 칸은 1로 주어지며, 각 숫자 사이에는 빈칸이 하나씩 있다.

www.acmicpc.net

분할 정복 문제로, 재귀로 풀면 된다.

#include <iostream>
#include <vector>
using namespace std;
vector<vector<int>> v;

int white = 0;
int blue = 0;

void divide(int startRow, int startCol, int endRow, int endCol) {
	//cout << startRow << "~" << endRow << "행 " << startCol << "~" << endCol << "열" << endl;
	int color = v[startRow][startCol];
	bool flag = true;
	for (int i = startRow; i <= endRow; i++) {
		for (int j = startCol; j <= endCol; j++) {
			if (v[i][j] != color) {
				flag = false;
				break;
			}
		}
		if(!flag)
			break;
	}
	if (!flag) {
		//cout << "한 덩어리로 안 됨" << endl;
		divide(startRow, startCol, startRow + (endRow - startRow) / 2, startCol + (endCol - startCol) / 2);
		divide(startRow, startCol + (endCol - startCol) / 2 + 1, startRow + (endRow - startRow) / 2, endCol);
		divide(startRow + (endRow - startRow) / 2 + 1, startCol, endRow, startCol + (endCol - startCol) / 2);
		divide(startRow + (endRow - startRow) / 2 + 1, startCol + (endCol - startCol) / 2 + 1, endRow, endCol);
	}
	else {
		if (v[startRow][startCol] == 0) {
			//cout << "흰 종이" << endl;
			white++;
		}
		else {
			//cout << "파란 종이" << endl;
			blue++;
		}
	}
}

int main() {
	cin.tie(NULL);
	ios_base::sync_with_stdio(false);
	int n;
	cin >> n;
	v = vector<vector<int>>(n, vector<int>(n));
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			cin >> v[i][j];
		}
	}
	divide(0, 0, n - 1, n - 1);
	cout << white << endl;
	cout << blue << endl;
	return 0;
}
728x90

'알고리즘 문제' 카테고리의 다른 글

[백준] 4949번 균형잡힌 세상  (0) 2020.04.08
[백준] 2589번 보물섬  (0) 2020.04.07
[백준] 5086번 배수와 약수  (0) 2020.03.30
[백준] 10996번 별 찍기 - 21  (0) 2020.03.30
[백준] 10039번 평균 점수  (0) 2020.03.29

+ Recent posts