力扣338. 比特位计数（动态规划）

力扣338. 比特位计数（动态规划）

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

示例 1:

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

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

给出时间复杂度为O(n*sizeof(integer))的解答非常容易。但你可以在线性时间O(n)内用一趟扫描做到吗？
要求算法的空间复杂度为O(n)。
你能进一步完善解法吗？要求在C++或任何其他语言中不使用任何内置函数（如 C++ 中的 __builtin_popcount）来执行此操作。

 

常规思路，直接计算

思路

复杂度分析

时间复杂度：O(k×num)，其中 k 是 int 型的二进制位数，k=32。需要对从 0 到 num 的每个数使用 O(k) 的时间计算「一比特数」，因此时间复杂度是 O(k×num)。

空间复杂度：O(1)。除了返回的数组以外，空间复杂度为常数。

#include <iostream>
#include<vector>
#include<math.h>
using namespace std;

class Solution {
public:
    vector<int> countBits(int num) {
        vector<int> result(num + 1,0);
        for (int i = 0; i <=num; i++)
        {
            result[i] = howone(i);
        }
        return result;
    }
    int howone(int num)
    {
        int result = 0;
        for (int n = 64; n >=0; n--)
        {
            if (num>=pow(2,n))
            {
                num = num - pow(2, n);
                result++;
            }
        }
        return result;
    }
};
int main()
{
    Solution s;
    auto result = s.countBits(20);
    for (int i = 0; i < result.size(); i++)
    {
        cout << result[i] << " ";
    }
    cout << endl;
    std::cout << "Hello World!\n";
}

 

动态规划（对应官方题解的方法三，最低有效位）

思路

对于所有的数字，只有两类：

奇数：二进制表示中，奇数一定比前面那个偶数多一个 1，因为多的就是最低位的 1。
          举例： 
         0 = 0       1 = 1
         2 = 10      3 = 11
偶数：二进制表示中，偶数中 1 的个数一定和除以 2 之后的那个数一样多。因为最低位是 0，除以 2 就是右移一位，也就是把那个 0 抹掉而已，所以 1 的个数是不变的。
           举例：
          2 = 10       4 = 100       8 = 1000
          3 = 11       6 = 110       12 = 1100
另外，0 的 1 个数为 0，于是就可以根据奇偶性开始遍历计算了

复杂度分析

时间复杂度：O(num)。对于每个数，只需要 O(1) 的时间计算「一比特数」。

空间复杂度：O(1)。除了返回的数组以外，空间复杂度为常数。

class Solution {
public:
    vector<int> countBits(int num) {
        vector<int> result(num + 1,0);
        result[0] = 0;
        for (int i = 1; i <=num; i++)
        {
            if (i%2==1)
            {
                result[i] = result[i - 1] + 1;
            }
            else
            {
                result[i] = result[i / 2];
            }
        }
        return result;
    }
};
int main()
{
    Solution s;
    auto result = s.countBits(20);
    for (int i = 0; i < result.size(); i++)
    {
        cout << result[i] << " ";
    }
    cout << endl;
    std::cout << "Hello World!\n";
}