[백준 1780] 종이의 개수 with Node.js

문제 링크

https://www.acmicpc.net/problem/1780

1780번: 종이의 개수

풀이

const input = require("fs").readFileSync("/dev/stdin").toString().trim().split("\n");
const n = +input[0];
const paper = input.slice(1).map(v => v.split(" ").map(vv => +vv));

const countPaper = n => {
    const count = [0, 0, 0];
    const recursion = (n, x, y) => {
        const num = paper[y][x];
        let numCount = 0;
        for (let i=0; i<n; i++) {
            for (let j=0; j<n; j++) {
                if (paper[y+j][x+i] === num) numCount++;
                else break;
            }
        }
        if (numCount === n*n) count[num+1]++;
        else {
            n /= 3;
            for (let i=0; i<3; i++) {
                for (let j=0; j<3; j++) {
                    recursion(n, x+i*n, y+j*n);
                }
            }
        }
    }
    recursion(n, 0, 0);
    console.log(count.join("\n"));
};

countPaper(n);

이 문제는 [백준 2630] 색종이 만들기 문제와 거의 동일한 문제다. 

 

recursion에서 현재 종이에 적힌 숫자가 모두 같은 숫자면 해당 숫자의 카운트를 증가시키고, 그렇지 않으면 종이를 9분할하여 재귀호출하였다.