leetcode 338. Bit位计数

发表于 | 分类于

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

示例：
比如给定 num = 5 ，应该返回 [0,1,1,2,1,2].

进阶：

  • 给出时间复杂度为O(n * sizeof(integer)) 的解答非常容易。 但是你可以在线性时间O(n)内用一次遍历做到吗？
  • 要求算法的空间复杂度为O(n)。

  • 你能进一步完善解法吗？ 在c ++或任何其他语言中不使用任何内置函数（如c++里的 __builtin_popcount）来执行此操作。

  • my solution

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
class Solution(object):
    def countBits(self, num):
        """
        :type num: int
        :rtype: List[int]
        """
        def countv(v):
            vlen = 0
            while v:
                v &= (v-1)
                vlen += 1
            return vlen
        res = [0]*(num+1)
        for i in range(0,num+1,4):
            vlen = countv(i)
            res[i] = vlen
            if i + 1 > num:break
            res[i+1] = vlen+1
            if i + 2 > num:break
            res[i+2] = vlen+1
            if i + 3 > num:break
            res[i+3] = vlen+2
        return res
  • a faster solution 
1
2
3
4
5
6
7
8
9
10
11
12
13
class Solution(object):
    def countBits(self, num):
        """
        :type num: int
        :rtype: List[int]
        """
        if num == 0:return [0]
        if num == 1:return [0,1]
        res = [0]*(num+1)
        res[1] = 1
        for i in range(2,num+1):
            res[i] = res[i>>1] + i%2
        return res