PS/프로그래머스

[JS][구현]쿼드압축 후 개수 세기

returnzero 2023. 9. 24. 18:29
function solution(arr) {
    let result = 0
    let numberOfZero = 0
    // INPUT [[1,1,0,0],[1,0,0,0],[1,0,0,1],[1,1,1,1]]
    // OUPUT [[1,1],[1,0]], [[0,0],[0,0]], [[1,0],[1,1]], [[0,1][1,1]]
    // INPUT [[1,0][0,1]]
    // OUTPUT [[1],[0],[0],[1]]
    function toSmallerSquare(arr) {
        if (isFilledWith(arr, 0)) {
            numberOfZero += 1
            return
        }
        if (isFilledWith(arr, 1)) {
            result += 1
            return
        }
        if (arr.length == 1) {
            if (arr[0] == 1) {
                result += 1
            } else {
                numberOfZero += 1
            }
            return
        }

        let meta = []
        let ans = []

        let l = arr.length
        let k = arr.length / 2
        for (let i = 0; i < k; i++) {
            let temp = []
            for (let j = 0; j < k; j++) {
                temp.push(arr[i][j])
            }
            ans.push(temp)
        }
        meta.push(ans)
        ans = []
        for (let i = k; i < l; i++) {
            let temp = []
            for (let j = 0; j < k; j++) {
                temp.push(arr[i][j])
            }
            ans.push(temp)
        }
        meta.push(ans)
        ans = []
        for (let i = 0; i < k; i++) {
            let temp = []
            for (let j = k; j < l; j++) {
                temp.push(arr[i][j])
            }
            ans.push(temp)
        }
        meta.push(ans)
        ans = []
        for (let i = k; i < l; i++) {
            let temp = []
            for (let j = k; j < l; j++) {
                temp.push(arr[i][j])
            }
            ans.push(temp)
        }
        meta.push(ans)
        return meta.map(e => toSmallerSquare(e))
    }

    toSmallerSquare(arr)
    return [numberOfZero, result];
}


// INPUT [[1],[1],[1],[1]] , number = 1
// OUTPUT TRUE
function isFilledWith(arr, number) {
    for (let i = 0; i < arr.length; i++) {
        for (let j = 0; j < arr[0].length; j++) {
            if (arr[i][j] != number) return false
        }
    }
    return true
}

한시간쯤 걸렸다..

큰 정사각형을 네 개로 쪼개야 한다고 생각했다. 그리고 그게 재귀적으로 반복할거라고 생각했다. 금방 못풀꺼라 생각했는데 풀었다..