N×N크기의 행렬로 표현되는 종이가 있다. 종이의 각 칸에는 -1, 0, 1 중 하나가 저장되어 있다. 우리는 이 행렬을 다음과 같은 규칙에 따라 적절한 크기로 자르려고 한다.
이와 같이 종이를 잘랐을 때, -1로만 채워진 종이의 개수, 0으로만 채워진 종이의 개수, 1로만 채워진 종이의 개수를 구해내는 프로그램을 작성하시오.
첫째 줄에 N(1 ≤ N ≤ 37, N은 3k 꼴)이 주어진다. 다음 N개의 줄에는 N개의 정수로 행렬이 주어진다.
첫째 줄에 -1로만 채워진 종이의 개수를, 둘째 줄에 0으로만 채워진 종이의 개수를, 셋째 줄에 1로만 채워진 종이의 개수를 출력한다.
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
public class BOJ_1780 {
static int[][] papers;
static int[] answers = new int[3];
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
int N = Integer.parseInt(br.readLine());
papers = new int[N][N];
for (int i = 0; i < N; i++) {
String[] str = br.readLine().split(" ");
for (int j = 0; j < N; j++) {
papers[i][j] = Integer.parseInt(str[j]);
}
}
partition(0, 0, N);
System.out.println(answers[0]); // -1
System.out.println(answers[1]); // 0
System.out.println(answers[2]); // +1
}
public static void partition(int row, int col, int size) {
if (check(row, col, size)) {
if (papers[row][col] == -1) ++answers[0];
else if (papers[row][col] == 0) ++answers[1];
else ++answers[2];
} else {
int newSize = size / 3;
for (int i = 0; i < 3; i++) {
for (int j = 0; j < 3; j++) {
partition(row + (newSize * i), col + (newSize * j), newSize);
}
}
}
}
public static boolean check(int row, int col, int size) {
int check = papers[row][col];
for (int i = row; i < row + size; i++) {
for (int j = col; j < col + size; j++) {
if (check != papers[i][j]) return false;
}
}
return true;
}
}
정답을 출력하기 위해 크기가 3인 answers 배열을 생성했다.
한 개의 숫자로만 채워진 종이의 수를 구하고 만약 한 개의 숫자로 채워지지 않은 종이라면 다시 9등분하여 재귀 호출할 partition 메소드를 생성했다.
또한 잘린 종이가 한 개의 숫자로만 채워져있는지 확인할 check 메소드를 생성했다. check 메소드에서 한 개의 숫자로만 채워진 종이라는 사실이 true로 밝혀지면, partition 메소드 조건문 내에서 해당 정답의 인덱스를 1 증가시켰다.