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

문제 링크

---

1. 아이디어

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

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

---

2. 코드

1. 풀이 [Java]

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

public class Main {

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

        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]);
        System.out.println(cnt[2]);
    }

    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 + 1]++;
        } else {
            int mr1 = sr + (er - sr) / 3;
            int mc1 = sc + (ec - sc) / 3;
            int mr2 = sr + (er - sr) / 3 * 2;
            int mc2 = sc + (ec - sc) / 3 * 2;

            dfs(sr, sc, mr1, mc1);
            dfs(sr, mc1, mr1, mc2);
            dfs(sr, mc2, mr1, ec);
            dfs(mr1, sc, mr2, mc1);
            dfs(mr1, mc1, mr2, mc2);
            dfs(mr1, mc2, mr2, ec);
            dfs(mr2, sc, er, mc1);
            dfs(mr2, mc1, er, mc2);
            dfs(mr2, mc2, er, ec);
        }
    }
}

2. 풀이 [C++]

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

int grid[2187][2187];
int cnt[3];

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] + 1]++;
    } else {
        int mr1 = sr + (er - sr) / 3;
        int mc1 = sc + (ec - sc) / 3;
        int mr2 = sr + (er - sr) / 3 * 2;
        int mc2 = sc + (ec - sc) / 3 * 2;

        dfs(sr, sc, mr1, mc1);
        dfs(sr, mc1, mr1, mc2);
        dfs(sr, mc2, mr1, ec);
        dfs(mr1, sc, mr2, mc1);
        dfs(mr1, mc1, mr2, mc2);
        dfs(mr1, mc2, mr2, ec);
        dfs(mr2, sc, er, mc1);
        dfs(mr2, mc1, er, mc2);
        dfs(mr2, mc2, 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';
    cout << cnt[2] << '\n';
}

---

3. 디버깅

없음.

---

4. 참고

없음.

---