LeetCode刷题之路：338. 比特位计数

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

示例 1:

输入: 2
输出: [0,1,1]
示例 2:

输入: 5
输出: [0,1,1,2,1,2]
进阶:

#最直观的思路

class Solution:
    def countBits(self, n: int) -> List[int]:
        # res = []
        # for i in range(n+1):
        #     count = 0
        #     while i > 0:
        #         count += (i & 1)
        #         i >>= 1
        #     res.append(count)
        # return res

        #超秒的方法
        res = [0]
        for i in range(1, n + 1):
            if i & 1:
                res.append(res[-1] + 1)
            else:
                res.append(res[int(i/2)])
        return res