[백준]_1780_종이의 개수

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class B1780 {
	static int N;
	static int[] count = new int[3];
	static int[][] arr;
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder();
		StringTokenizer st;
		
		N = Integer.parseInt(br.readLine());
		arr = new int[N][N];
		for(int i=0; i<N;i++) {
			st = new StringTokenizer(br.readLine());
			for(int j=0; j<N;j++) {
				arr[i][j] = Integer.parseInt(st.nextToken());
			}
		}
		
		divide(0,0,N);
		for(int i=0; i<3;i++) {
			sb.append(count[i]).append("\n");
		}
		System.out.println(sb);
	}
	
	private static void divide(int r, int c, int range) {
		// 조건 만족 시 종이 개수 카운트
		if(isSame(r,c,range)) {
			count[arr[r][c]+1]++;
			return;
		}
		
		// 조건을 만족하지 못했을 때 3 * 3으로 나누기 
		int three = range/3;
		for(int i=0; i<3;i++) {
			for(int j=0; j<3;j++) {
				divide(r + three*i,c + three*j,three);
			}
		}
	}
	// 범위 내 종이가 같은지 체크
	private static boolean isSame(int r, int c, int range) {
		int one = arr[r][c];
		for(int i=r; i<r+range;i++) {
			for(int j=c; j<c+range;j++) {
				if(arr[i][j]!=one) return false;
			}
		}
		return true;
	}
}