[BaekJoon] 1780번 - 종이의 개수 [Java][C++]

문제 링크

---

1. 문제 풀이

$N \times N$ 크기의 종이에 대해 현재 바라보는 종이가 전부 같은 숫자가 아니면 9등분을 한 후 다시 현재 바라보는 종이가 전부 같은 숫자인지 판단하며 같은 숫자면 해당 숫자의 개수를 세는 문제다. 현재 바라보는 종이의 영역을 활용한 재귀 함수로 해결할 수 있다.

재귀 함수는 현재 바라보는 종이의 맨 왼쪽 윗 칸의 위치 정보와 맨 오른쪽 아랫 칸의 위치 정보를 받아서 현재 종이가 전부 같은 숫자면 개수를 세고 종료하고 같은 숫자가 아니면 9등분한 각 영역에 대해 재귀를 타면 된다.

---

2. 코드

1. 풀이 [Java]

import java.io.*;
import java.util.*;

public class Main {

    static int[] cntArr = new int[3];

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;

        int N = Integer.parseInt(br.readLine());
        int[][] map = new int[N][N];
        for (int i = 0; i < N; i++) {
            st = new StringTokenizer(br.readLine());

            for (int j = 0; j < N; j++) {
                map[i][j] = Integer.parseInt(st.nextToken());
            }
        }

        recur(map, 0, 0, N, N);

        System.out.println(cntArr[0]);
        System.out.println(cntArr[1]);
        System.out.println(cntArr[2]);
    }

    static void recur(int[][] map, int r1, int c1, int r2, int c2) {
        boolean flag = true;
        int color = map[r1][c1];

        out:
        for (int i = r1; i < r2; i++) {
            for (int j = c1; j < c2; j++) {
                if (map[i][j] != color) {
                    flag = false;
                    break out;
                }
            }
        }

        if (flag) {
            cntArr[color + 1]++;
        } else {
            int mr1 = r1 + (r2 - r1) / 3;
            int mc1 = c1 + (c2 - c1) / 3;
            int mr2 = r1 + (r2 - r1) / 3 * 2;
            int mc2 = c1 + (c2 - c1) / 3 * 2;

            recur(map, r1, c1, mr1, mc1);
            recur(map, r1, mc1, mr1, mc2);
            recur(map, r1, mc2, mr1, c2);
            recur(map, mr1, c1, mr2, mc1);
            recur(map, mr1, mc1, mr2, mc2);
            recur(map, mr1, mc2, mr2, c2);
            recur(map, mr2, c1, r2, mc1);
            recur(map, mr2, mc1, r2, mc2);
            recur(map, mr2, mc2, r2, c2);
        }
    }
}

2. 풀이 [C++]

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

int board[2187][2187];
int cntArr[3];

bool check(int r1, int c1, int r2, int c2) {
    for (int i = r1; i < r2; i++) {
        for (int j = c1; j < c2; j++) {
            if (board[i][j] != board[r1][c1]) return false;
        }
    }

    return true;
}

void recur(int r1, int c1, int r2, int c2) {
    if (check(r1, c1, r2, c2)) {
        cntArr[board[r1][c1] + 1]++;
    } else {
        int mr1 = r1 + (r2 - r1) / 3;
        int mc1 = c1 + (c2 - c1) / 3;
        int mr2 = r1 + (r2 - r1) / 3 * 2;
        int mc2 = c1 + (c2 - c1) / 3 * 2;

        recur(r1, c1, mr1, mc1);
        recur(r1, mc1, mr1, mc2);
        recur(r1, mc2, mr1, c2);
        recur(mr1, c1, mr2, mc1);
        recur(mr1, mc1, mr2, mc2);
        recur(mr1, mc2, mr2, c2);
        recur(mr2, c1, r2, mc1);
        recur(mr2, mc1, r2, mc2);
        recur(mr2, mc2, r2, c2);
    }
}

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

    int n;
    cin >> n;

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

    recur(0, 0, n, n);

    cout << cntArr[0] << '\n';
    cout << cntArr[1] << '\n';
    cout << cntArr[2] << '\n';
}

---