<백준> 1780

진기명기·2025년 4월 26일

코딩테스트<C++>

목록 보기
63/212

종이의 개수

문제
N×N크기의 행렬로 표현되는 종이가 있다. 종이의 각 칸에는 -1, 0, 1 중 하나가 저장되어 있다. 우리는 이 행렬을 다음과 같은 규칙에 따라 적절한 크기로 자르려고 한다.
1. 만약 종이가 모두 같은 수로 되어 있다면 이 종이를 그대로 사용한다.
2. (1)이 아닌 경우에는 종이를 같은 크기의 종이 9개로 자르고, 각각의 잘린 종이에 대해서 (1)의 과정을 반복한다.
이와 같이 종이를 잘랐을 때, -1로만 채워진 종이의 개수, 0으로만 채워진 종이의 개수, 1로만 채워진 종이의 개수를 구해내는 프로그램을 작성하시오.

입력
첫째 줄에 N(1 ≤ N ≤ 37, N은 3k 꼴)이 주어진다. 다음 N개의 줄에는 N개의 정수로 행렬이 주어진다.

출력
첫째 줄에 -1로만 채워진 종이의 개수를, 둘째 줄에 0으로만 채워진 종이의 개수를, 셋째 줄에 1로만 채워진 종이의 개수를 출력한다.

앞의 문제 하얀색, 파란색 종이 숫자 세는 거와 비슷한 문제로, 9로 나눠지는 부분을 고려하면 쉽게 풀 수 있는 문제였다

int board[2200][2200];
int plus1 = 0;
int zero = 0;
int minus1 = 0;

void divide(int x, int y, int size)
{
	int origin = board[x][y];
	bool same = true;

	for (int i = x; i < x + size; i++)
	{
		for (int j = y; j < y + size; j++)
		{
			if (board[i][j] != origin)
			{
				same = false;
				break;
			}
		}
		if (!same)
			break;
	}
	if (same)
	{
		if (origin == 0)
			zero++;
		else if (origin == 1)
			plus1++;
		else
			minus1++;
	}
	else
	{
		for (int i = 0; i < 3; i++)
		{
			for (int j = 0; j < 3; j++)
			{
				int newSize = size / 3;
				divide(x + i * newSize, y + j * newSize, newSize);
			}
		}
	}
	
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	int n;
	cin >> n;

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

	divide(0, 0, n);
	
	cout << minus1 << "\n";
	cout << zero << "\n";
	cout << plus1 << "\n";
}

0개의 댓글