(c++) 백준 "1780) 종이의 개수"

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

1780번: 종이의 개수

---

#include <iostream>
#include <vector>
#include <string>
using namespace std;
int main() {
	int N; cin >> N;
	vector<vector<int>> board;
	for (int i = 0; i < N; i++) {
		vector<int> tmp;
		for (int k = 0; k < N; k++) {
			int n; cin >> n;
			tmp.push_back(n);
		}
		board.push_back(tmp);
	}
	vector<int> answer(4);
	while (board.size() >= 3) {
		vector<vector<int>> list;
		for (int i = 0; i < board.size() / 3; i++) {
			vector<int> tmp;
			for (int k = 0; k < board.size() / 3; k++) {
				int count[4] = { 0 };
				for (int j = 0; j < 3; j++) {
					for (int t = 0; t < 3; t++) {
						count[(board[(3 * i) + j][(3 * k) + t])+1]++;
					}
				}
				if (count[3] > 0) {
					for (int j = 0; j < 3; j++) {
						answer[j] += count[j];
					}
					tmp.push_back(2);
				}
				else if (count[0] == count[1] && count[1] == 0) {
					tmp.push_back(1);
				}
				else if (count[1] == count[2] && count[2] == 0) {
					tmp.push_back(-1);
				}
				else if (count[0] == count[2] && count[2] == 0) {
					tmp.push_back(0);
				}
				else {
					for (int j = 0; j < 3; j++) {
						answer[j] += count[j];
					}
					tmp.push_back(2);
				}
			}
			list.push_back(tmp);
		}
		board.assign(list.begin(), list.end());
	}
	answer[board[0][0] + 1]++;
	for (int i = 0; i < 3; i++) {
		cout << answer[i] << endl;
	}
	return 0;
}