Haribo ML, AI, MATH, Algorithm

쿼드압축 후 개수 세기


쿼드압축 후 개수 세기

코드

import numpy as np

def solution(arr):

    def fn(a):
        if np.all(a == 0): return np.array([1, 0])
        if np.all(a == 1): return np.array([0, 1])
        n = a.shape[0]//2
        return fn(a[:n, :n]) + fn(a[n:, :n]) + fn(a[:n, n:]) + fn(a[n:, n:])

    return fn(np.array(arr)).tolist()

Similar Posts

이전 포스트 카펫

다음 포스트 타겟 넘버

Comments