写在前面:
最近一直在做一些LeetCode的题目,我之前一直是将每道题我的答案和想法以及别人优秀的答案都整合在一起然后上传的GitHub上,但是我发现这样做自己还是很难去经常复习,尤其最近发现很多之前遇到的问题再次发生。所以很有必要对一些阶段性的问题进行归纳和总结。
今天遇到的是这样的题目:
题目非常的简单,就是要求输入整数的对应二进制序列的1的个数,要求o(n)的时间复杂度和o(n)的空间复杂度。
废话不多说我的答案如下:
class Solution {
public:
vector<int> countBits(int num) {
vector<int> ans;
for(int i=0;i<=num;i++) {
int count = 0;
int j = i;
while(j) {
count += j%2;
j = j >> 1;//问题
}
ans.push_back(count);
}
return ans;
}
};
问题出在注释的那一行,这是不知道第几次出现>>移位操作符的时候我想当然把这个当做是对j操作,其实不是他是返回了一个临时的变量,而原本的j并没有改变,所以必须要将这个值赋值回给j。
更好的方案:
class Solution {
public:
vector<int> countBits(int num) {
vector<int> ret(num+1, 0);
for (int i = 1; i <= num; ++i)
ret[i] = ret[i&(i-1)] + 1;
return ret;
}
};
我认为这个做法是类似一位动态规划。
划分其实很简单,将前面的结果缓存起来,用于后面的求解。
关键在于这个式子的理解:ret[i] = ret[i&(i-1)] + 1;
i&(i-1) 是i和i-1对应二进制序列中相同部分的二进制序列,记这个序列中1的位数为a,那么i中一的位数必然等于a+1。