LeetCode 338. Counting Bits

比较容易想到的就是一个个数拿出来判断有几个1,判断时可以采用n&(n-1)来简化过程,不过时间复杂度明显就是O(n*sizeof(integer)).。

此外可以采用动态规划来做,对于一个大于1的整数,其二进制至少有2位,假设是n位,可以把它分成前(n-1)位和最后1位分别统计。

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

整体时间复杂度是O(n)

你可能感兴趣的:(LeetCode 338. Counting Bits)