(M)Dynamic Programming:338. Counting Bits

最新推荐文章于 2023-10-09 02:02:27 发布

努力努力再努力r

最新推荐文章于 2023-10-09 02:02:27 发布

阅读量128

点赞数

分类专栏： leetcode DP

本文链接：https://blog.csdn.net/u012528000/article/details/78128695

版权

leetcode 同时被 2 个专栏收录

264 篇文章 0 订阅

订阅专栏

46 篇文章 0 订阅

订阅专栏

肯定不能把每一个数的每一位都找一遍。先写出几个来找规律，发现如果求num=i+1这个数字1的个数：

if(i最后一位是0) i中1的个数+1

else if(i倒数第2位是0) i中1的个数+0

else if(i的倒数第3位是0) i中1的个数-1

else if(i的倒数第4位是0) i中1的个数-2

……

else if(i的倒数第n位是0) i中1的个数-(n-2)

class Solution {
public:
    vector<int> countBits(int num) {
        vector<int> dp(num+1);
        dp[0] = 0;
        if(num >= 1)
            dp[1] = 1;
        for(int i = 2; i < num + 1; ++i)
        {
            int count = 1;
            while(((i-1) >> (count-1)) & 0x01)
            {
                count++;
            }
            //count表示第几位是0（从1开始数）
            dp[i] = dp[i - 1] - (count - 2);
        }
        return dp;
    }
};

注意怎么求倒数第n+1位：(i >> n) & 1，而i & (1 >> n)是取正数第n+1位。

我的方法比较笨，别人都发现了更好的规律，解法很多，这里粘两个。其中一个巧妙用了 i & (i - 1)：

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

这个本来是用来判断一个数是否是2的指数的快捷方法，比如8，二进制位1000, 那么8&(8-1)为0，只要为0就是2的指数。找规律发现每个i值都是i&(i-1)对应的值加1。

还有一个方法是不提倡的，但是要会用，直接返回有几个1：

class Solution {
public:
    vector<int> countBits(int num) {
        vector<int> res;
        for (int i = 0; i <= num; ++i) {
            res.push_back(bitset<32>(i).count());
        }
        return res;
    }
};