题目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)2] = ans[0]+1
+ ans[(11)2] = ans[1]+1
+ ans[(110)2] = ans[(10)2] + 1
没错!每一个数n的二进制中1的个数等于(n-最接近n的2的整次幂)的二进制1的个数+1
所以这里,我们可以将问题用动态规划。
状态转移方程如下:
d(n) = d(n-nearest2^n)+1
最终提交打败了95.7%的上传者。