338. 比特位计数简单 DP

Posted 2021-12-07 幽殇默

https://leetcode-cn.com/problems/counting-bits/
很容易可以想到一个暴力的做法

class Solution {
public:
    vector<int> countBits(int n) 
    {
        vector<int>ans;
        for(int i=0;i<=n;i++)
        {
            int temp=i,cnt=0;
            while(temp) temp-=lowbit(temp),cnt++;
            ans.push_back(cnt);
        }
        return ans;
    }   
    int lowbit(int x) 
    {
        return x&-x;
    }
};

考虑DP做法。
状态转移f[i]=f[i>>1]+(i&1)

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