题目:Counting Bits
Given a non negative integer number num. For every numbers i in the range 0 <= i <= num calculate the number of 1’s in their binary representation and return them as an array.
Example:For num = 5 you should return [0,1,1,2,1,2].
原题链接
简单翻译一下,就是统计从0到num的所有数字的二进制形式下1的个数。
当时看到这个题目的时候第一个想法是开了一个数组来模拟二进制的进位情况,事实证明可行但是效率不高,后来网上查了一下,下面这个解法比较好,先贴代码:
class Solution{
public:
vector<int> countBits(int num) {
vector<int> ans = {0};
int lastPow;
for(int i =1; i <= num; ++i){
if((i & (i - 1)) == 0){
ans.push_back(1);
lastPow = i;
}else{
ans.push_back(1 + ans[i - lastPow]);
}
}
return ans;
}
};
简单说一下原理,先给出一个从0到7(二进制)的例子:
十进制 | 二进制 |
---|---|
0 | 0 |
1 | 1 |
2 | 10 |
3 | 11 |
4 | 100 |
5 | 101 |
6 | 110 |
7 | 111 |
仔细对比一下,0到3的二进制部分恰好可以看成4到7的“后半部分”,唯一的区别就是最前面多了一位1,所以这个可以给我们一个启发,利用前面的数字中的1来“递归”的计算后面的数字中1的个数,例子中4到7中二进制1的个数就可以看到是对应的0到3中二进制1的个数加1,那么,怎么区分什么时候开始新一轮“递归”呢,分隔线就是在2的整数幂,所以可以每当统计到一个2的整数幂的时候,把之前统计的数字的结果都作为新一轮的“右半部分”来直接计算,用代码描述就是:
ans.push_back(1 + ans[i - lastPow]);
这之前还有一个小技巧,就是关于怎么判断一个正整数是不是2的整数次幂,很简单,2的整数次幂就是第一位是1后面都是0,而比它小1的数字正好是后面都是1而前面是0,可以把他们相与,结果是0的话就说明是2的整数幂:
if((i & (i-1)) == 0) //为真则i是2的整数幂
最后,在网上看到一个精简版本的代码,顺便一起贴出来:
vector<int> countBits(int num) {
vector<int> bits(num+1,0);
for(int i = 1; i <= num; ++i)
bits[i] += bits[i & (i-1)] + 1;
return bits;
}