338. 比特位计数(Python)

338. 比特位计数

给定一个非负整数 num。对于 0 ≤ i ≤ num 范围中的每个数字 i ，计算其二进制数中的 1 的数目并将它们作为数组返回。

示例 1:

输入: 2

输出: [0,1,1]

示例 2:

输入: 5

输出: [0,1,1,2,1,2]

def func(num):
    dp = [0]*(num+1)
    for i in range(1, num+1):
        dp[i] = dp[i & (i-1)]+1
    # for i in range(num+1):
        # dp[i] = bin(i).count('1')
    return dp

print(func(5))