백준 - 1992 쿼드트리

Park Inhye·2024년 4월 11일

코테 연습

목록 보기

34/37

문제

흑백 영상을 압축하여 표현하는 데이터 구조로 쿼드 트리(Quad Tree)라는 방법이 있다. 흰 점을 나타내는 0과 검은 점을 나타내는 1로만 이루어진 영상(2차원 배열)에서 같은 숫자의 점들이 한 곳에 많이 몰려있으면, 쿼드 트리에서는 이를 압축하여 간단히 표현할 수 있다.

주어진 영상이 모두 0으로만 되어 있으면 압축 결과는 "0"이 되고, 모두 1로만 되어 있으면 압축 결과는 "1"이 된다. 만약 0과 1이 섞여 있으면 전체를 한 번에 나타내지를 못하고, 왼쪽 위, 오른쪽 위, 왼쪽 아래, 오른쪽 아래, 이렇게 4개의 영상으로 나누어 압축하게 되며, 이 4개의 영역을 압축한 결과를 차례대로 괄호 안에 묶어서 표현한다

위 그림에서 왼쪽의 영상은 오른쪽의 배열과 같이 숫자로 주어지며, 이 영상을 쿼드 트리 구조를 이용하여 압축하면 "(0(0011)(0(0111)01)1)"로 표현된다. N ×N 크기의 영상이 주어질 때, 이 영상을 압축한 결과를 출력하는 프로그램을 작성하시오.

입력

첫째 줄
- 영상의 크기를 나타내는 숫자 N
두 번째 줄부터
- 영상의 각 점

제한 사항

N 은 언제나 2의 제곱수, 1 ≤ N ≤ 64
각 문자열은 0 또는 1의 숫자

예시

// NOTE: 입력
8
11110000
11110000
00011100
00011100
11110000
11110000
11110011
11110011

// NOTE: 출력
((110(0101))(0010)1(0001))

해결

백준 - 1780 종이의 개수 문제를 응용해서 풀었다!

종이의 개수 문제와 비교

탐색 범위가 2X2로 나눈다.
재귀 전, return 값 앞에 '('를 붙여준다.
재귀 후, return 값 뒤에 ')'를 붙여준다.

꼭! 재귀 전에 여는 소괄호('(')를 붙여야 한다.
return '(' + 재귀실행값 + ')'로 잘못 설정했다가,
모든 수가 소괄호에 갇혔었다.🥹

전체 코드

const [n, ...inputs] = require('fs')
    .readFileSync('dev/stdin')
    .toString()
    .trim()
    .split('\n');

const DIV = 2;
const N = +n;
const _data = inputs.map(v => [...v].map(v => +v));

const quadTree = (n) => {
    const divideArr = (matrixSize, startX, startY) => {
        // NOTE: 행렬의 기준
        const _baseNum = _data[startX][startY];

        // NOTE: 탐색 진행률
        let _progressCount = 0;

        // NOTE: 행렬 완전탐색
        for (let i = 0; i < matrixSize; i++) {
            for (let j = 0; j < matrixSize; j++) {
                if (_data[startX + i][startY + j] == _baseNum) _progressCount += 1;
                else break;
            }
        }

        // NOTE: 모든 숫자가 동일
        if (_progressCount == matrixSize ** 2) return _baseNum;
        
        // NOTE: 다른 숫자가 섞임 > 재탐색
        matrixSize /= DIV;
        let _result = '(';
        
        for (let i = 0; i < DIV; i++) {
            for (let j = 0; j < DIV; j++) {
                _result += divideArr(matrixSize, startX + matrixSize * i, startY + matrixSize * j);
            }
        }

        return _result + ')';
    }

    console.log(divideArr(n, 0, 0));
}

quadTree(N);