leetcode338比特位计数c++

题目描述

给定一个非负整数 num。对于 0 ≤ i ≤ num 范围中的每个数字 i ，计算其二进制数中的 1 的数目并将它们作为数组返回。
示例 1:
输入: 2
输出: [0,1,1]
示例 2:
输入: 5
输出: [0,1,1,2,1,2]

代码实现

动态规划的思想：首先要知道，n 与 n&(n-1) 的1的个数只差1；然后就可以从0开始，利用计算出的 0~ i-1 的计数结果，计算 i 的计数结果: N(i) = N( i & ( i-1) ) + 1。

为什么n 与 n&(n-1) 的1的个数只差1？
对二进制数观察，我们知道 n 与 n-1 的最末位一定不同，而且从 n-1 到 n 有一定的变化规律：

1. 若是 n - 1 的最后一位是 0，那么 n 的最后一位一定是 1，并且对于 1 的计数比 n-1 大 1，这个时候 n & (n-1) == n-1。
2. 若是 n - 1 的最后一位为 1，那么 n 的最后一位一定是 0。这个时候若 n-1 的前一位是0，则 n 的前一位是 1；若n-1 的前一位是1，则 n 的前一位是 0，继续往前进 1 。这个时候 n 的1的计数比 n & (n-1) 大1。

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