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）来执行此操作。

致谢：

特别感谢 @syedee 添加此问题及所有测试用例。

代码：

i    bin       '1'    i&(i-1)
 0  0000    0
-----------------------
 1  0001    1    0000
-----------------------
 2  0010    1    0000
 3  0011    2    0010
-----------------------
 4  0100    1    0000
 5  0101    2    0100
 6  0110    2    0100
 7  0111    3    0110
-----------------------
 8  1000    1    0000
 9  1001    2    1000
10  1010    2    1000
11  1011    3    1010
12  1100    2    1000
13  1101    3    1100
14  1110    3    1100
15  1111    4    1110

找规律：若i是偶数，则第i个数二进制中含1的数目等于第i/2二进制数中含1的数目；若i是奇数，则第i个数的二进制数中含1的个数等于第i/2二进制含1的个数加1；

所以代码:

class Solution {
public:
    vector<int> countBits(int num) {
        vector<int>res{0};
        for(int i=1;i<=num;i++){
            if(i%2==0)res.push_back(res[i/2]);
            else res.push_back(res[i/2]+1);
        }
        return res;
    }
};

方法二、

第i个数的二进制中数含1的个数等于等于第（i&i-1）个数中二进制数中1的个数加1；代码如下：

class Solution {
public:
    vector<int> countBits(int num) {
        vector<int>res{0};
        for(int i=1;i<=num;i++){
            res.push_back(res[i&(i-1)]+1);
        }
        return res;
    }
};