338. 比特位计数【简单 DP】

在这里插入图片描述
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;
    }   
};

猜你喜欢

转载自blog.csdn.net/bettle_king/article/details/121185696