[BaekJoon] 2630번 - 색종이 만들기 [Java][C++]

문제 링크

---

1. 문제 풀이

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

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

---

2. 코드

1. 풀이 [Java]

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

public class Main {

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

    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]);
    }

    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]++;
        } else {
            int mr = (r1 + r2) / 2;
            int mc = (c1 + c2) / 2;

            recur(map, r1, c1, mr, mc);
            recur(map, r1, mc, mr, c2);
            recur(map, mr, c1, r2, mc);
            recur(map, mr, mc, r2, c2);
        }
    }
}

2. 풀이 [C++]

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

int board[128][128];
int cntArr[2];

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]]++;
    } else {
        int mr = (r1 + r2) / 2;
        int mc = (c1 + c2) / 2;

        recur(r1, c1, mr, mc);
        recur(r1, mc, mr, c2);
        recur(mr, c1, r2, mc);
        recur(mr, mc, 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';
}

---