Description:
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].
Follow up:
It is very easy to come up with a solution with run time O(n*sizeof(integer)). But can you do it in linear time O(n) /possibly in a single pass?
Space complexity should be O(n).
Can you do it like a boss? Do it without using any builtin function like __builtin_popcount in c++ or in any other language.
描述:
给定一个正整数num,对于每一个在区间[0,num]的正整数i,算出 i 1的位数,返回一个包含这些位数的数组
约束:
1.O(n)的时间复杂度和空间复杂度
2.不能使用任何内建函数
分析:对于这种题目,肯定需要位运算来解决,我们可以使用下面这个式子来去掉数字二进制表示的最后一个1:
num = num & (num - 1)
比如说num = 6; 二进制为0110,那么 6&5 = 4 二进制为0100.
这样我们就可以得到第一个版本的代码:
/*
@param num:用户输入
*/
vector<int> result(num+1, 0);
for (int i = 1; i <= num; ++i) {
int cnt = 0;
int current = i;
while (current != 0) {
current = current & (current - 1);
cnt++;
}
result[i] = cnt;
}
return result;
下面会用到一些动态规划的知识:
1.重叠子问题:
我们不需要对每一个数字都进行一遍while循环,因为有很多数字的重复部分,我们可以利用之前的结果来直接索引到,比如说我们需要计算7有多少个1,那么我们只需要进行一次与操作:7&6=6;因为6的1我们已经计算出来了,并且我们去掉了7的最后一个1,所以我们需要把这个1加上.则可以得到下面的递归关系:
result[i] = result[i&(i-1)] +1;
利用动态规划的思想我们就可以简化第一段代码,把时间复杂度从O(CN)降低到O(N):
/*
@param num:用户输入
*/
vector<int> result(num+1, 0);
for (int i = 1; i <= num; ++i) {
result[i] = result[i&(i-1)] + 1;
}
return result;
总结:知识点为位运算和动态规划.
补:
问题:为什么num & (num - 1)可以去掉二进制的最后一个1?
自己理解:num-1可以改变二进制1位到最后1的那一位之间的所有状态,比如我们有一个二进制:
0101110000
那么我们把这个数字减1,可以得到:
0101101111
可以看到从第1位到第5位之间的二进制的状态都取反了,但是剩下的高位却不变,我们再把这个数跟原来的数进行与操作,那么就可以得到高位不变,但是低位全部清零的效果,这样我们就可以去掉二进制的最后一个1.