Leetcode: 338.Counting Bits

题目338.Counting Bit

Language:cpp
选上这道题是刚好听到老师在评价别的同学的博客的时候，提到了他用了Pick One随机选题，我觉得很有趣就来尝试了一下。题目难度为Medium。

题目中文简述：

给定一个输入num，输出一个num长度的vector，vector中每一位数为对应下标的二进制中的1的个数。
题目描述不难，首先先把sourcecode 放出来。

#include<cmath>
class Solution {
public:

    int nearx2(int n){
        int index = 0;
        while(n!=0){
            n /=2;
            index++;
        }
        return pow(2,index-1);
    }

    vector<int> countBits(int num) {
        vector<int> ans(num+1);
        ans[0]=0;
        if(num!=0)
            ans[1]=1;

        for(int i = 2; i <= num; i++){
            ans[i] = ans[i-nearx2(i)]+1; 
        }
        return ans;
    }
};

算法分析

看下代码，其实题目真的不难。最常见的算法就是把下标i用进制分解的算法，来统计i的二进制中1的个数。这样的话算法的时间复杂度就为O(nlogn)。

但是Leetcode中有一个进阶要求是使用非O(n*sizeof(integer))也就是我上面提到的O(nlogn)的方法，尝试使用线性级别O(n)的算法.首先我想到的就是打表的方法，直接将所有int的答案生成出来，然后取前num段，但是Leetcode还有第二个要求就是空间复杂度要为O(n).

经过一段时间的思考，简单列举出前8个数的二进制形式：0，1，10，11，100，101，110，111。发现有什么规律没有？
+ ans[(10)₂] = ans[0]+1
+ ans[(11)₂] = ans[1]+1
+ ans[(110)₂] = ans[(10)₂] + 1
没错！每一个数n的二进制中1的个数等于（n-最接近n的2的整次幂）的二进制1的个数+1
所以这里，我们可以将问题用动态规划。
状态转移方程如下：
d(n) = d(n-nearest2^n)+1

最终提交打败了95.7%的上传者。