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

문제 링크

---

1. 아이디어

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

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

---

2. 코드

1. 풀이 [Java]

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

public class Main {

    static int[][] grid;
    static int[] cnt = 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());

        grid = new int[n][n];
        for (int i = 0; i < n; i++) {
            st = new StringTokenizer(br.readLine());
            for (int j = 0; j < n; j++) {
                grid[i][j] = Integer.parseInt(st.nextToken());
            }
        }

        dfs(0, 0, n, n);

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

    static void dfs(int sr, int sc, int er, int ec) {
        boolean flag = true;
        int color = grid[sr][sc];

        out:
        for (int r = sr; r < er; r++) {
            for (int c = sc; c < ec; c++) {
                if (grid[r][c] != color) {
                    flag = false;
                    break out;
                }
            }
        }

        if (flag) {
            cnt[color]++;
        } else {
            int mr = (sr + er) / 2;
            int mc = (sc + ec) / 2;

            dfs(sr, sc, mr, mc);
            dfs(sr, mc, mr, ec);
            dfs(mr, sc, er, mc);
            dfs(mr, mc, er, ec);
        }
    }
}

2. 풀이 [C++]

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

int grid[128][128];
int cnt[2];

bool check(int sr, int sc, int er, int ec) {
    for (int r = sr; r < er; r++) {
        for (int c = sc; c < ec; c++) {
            if (grid[r][c] != grid[sr][sc]) return false;
        }
    }

    return true;
}

void dfs(int sr, int sc, int er, int ec) {
    if (check(sr, sc, er, ec)) {
        cnt[grid[sr][sc]]++;
    } else {
        int mr = (sr + er) / 2;
        int mc = (sc + ec) / 2;

        dfs(sr, sc, mr, mc);
        dfs(sr, mc, mr, ec);
        dfs(mr, sc, er, mc);
        dfs(mr, mc, er, ec);
    }
}

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

    int n;
    cin >> n;

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

    dfs(0, 0, n, n);

    cout << cnt[0] << '\n';
    cout << cnt[1] << '\n';
}

---

3. 디버깅

없음.

---

4. 참고

없음.

---