백준 2630 색종이 만들기 c++ // 재귀함수 , 일반식만들기

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

2630번: 색종이 만들기

1. 내 코드

#include <bits/stdc++.h>
using namespace std;

int n,ret1,ret2, a[150][150];

//시작좌표~끝좌표(시작좌표+n)까지 돌면서 모두같은지 검사
bool all_same(int n, int y, int x) {

	int first = a[y][x];
	for (int i = y; i < y + n; ++i) {
		for (int j = x; j < x + n; ++j) {
			if (a[i][j] != first) {
				return false;
			}
		}
	}

	return true;

}

//함수정의 : 크기n*n, 시작좌표 
void go(int n, int y, int x) {
	//기저사례
	if (all_same(n,y,x)) {

		if (a[y][x] == 0) {
			ret1++;
		}
		if (a[y][x] == 1) {
			ret2++;
		}

		return;
	}

	int half = n / 2;
	//재귀식
	//4칸으로 나눈후 각각칸의 시작좌표를 구했음.
	go(half, y, x);
	go(half, y, x+half);
	go(half, y+half, x);
	go(half, y+half, x+half);
}

int main() {
	ios_base::sync_with_stdio(0);
	cin.tie(0); cout.tie(0);

	cin >> n;

	for (int i = 0; i < n; ++i) {
		for (int j = 0; j < n; ++j) {
			cin >> a[i][j];
		}
	}

	go(n, 0, 0);

	cout << ret1 << "\n" << ret2 << "\n";

}