leetcode动态规划法求解338题比特位计数

最新推荐文章于 2021-12-15 22:12:04 发布

learing_

最新推荐文章于 2021-12-15 22:12:04 发布

阅读量139

点赞数 1

分类专栏： leetcoed刷题文章标签： leetcode 算法动态规划

本文链接：https://blog.csdn.net/a56895/article/details/114306363

版权

leetcoed刷题专栏收录该内容

2 篇文章 0 订阅

订阅专栏

比特位计数

题目介绍

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

示例 1:

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/counting-bits
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解法1（时间复杂度为n*num）

这种方法整体就是使用一个循环，循环遍历每一个数字，计算当前数字的1的个数，然后记录下来，最后返回整体。所以我们可以写一个函数来计算某一个数字有多少个1。

该函数其实整体比较简单，最为需要明白的一个点在于 x = x&(x-1) 这个表达式的含义，这个表达式会把x最后的1变成0。所以我们每进行一次这个计算，就加上1，一直到x变为0为止，最后累加得到的数字就是1的数目。

整体代码如下：

class Solution {
public:
    vector<int> countBits(int num) {
        vector<int> Bits_array;
        Bits_array.resize(num+1);
        Bits_array[0] = 0;
        for(int i = 1;i<=num;i++){
            Bits_array[i] = count_one(i);

        }
        return Bits_array;
    }
    int count_one(int num){
        int x = num;
        int one = 0;
        while(x>0){
            x = x&(x-1);
            one++;
        }
        return one;
    }
};

解法2（时间复杂度为n）

其实这个题目是一个十分经典的动态规划问题。一个二进制数字多加一个比特位，其实1的数字就是多了1个。如果我们知道当前数字对应的最大2的幂为y，这个数字为x,假设z=x-y。那么bit[x] = bit[z] + 1;这样问题就转换为了一个标准的动态规划问题。所以目前我们需要知道如何判断一个数字是否是2的幂。如果是，就记录当前最大2的幂，并且把当前的bit[x]记录为1。这样最终就得到了bit数组。

整体代码如下：

class Solution {
public:
    vector<int> countBits(int num) {
        vector<int> countBit_array(num+1);
        int max_bit = 0;
        for (int i = 1;i<=num;i++){
            if((i&(i-1)) == 0){
                max_bit = i;
                countBit_array[i] = 1;
            }
            else{
                countBit_array[i] = 1+countBit_array[i-max_bit];
            }
        }
        return countBit_array;
    }
};