题目:
给你一个整数 n
,对于 0 <= i <= n
中的每个 i
,计算其二进制表示中 1
的个数 ,返回一个长度为 n + 1
的数组 ans
作为答案。
分析:
首先很容易想到从0开始遍历到n,对于每个数字计算其二进制中1的个数然后放在ans数组中,其下标表示二进制对应的数字。所以该问题关键是如何计算每个数字二进制中1的个数。
计算二进制中1的个数一般可以使用部分编程语言的内置函数,如java的Integer.bitCount、C++的__builtin_popcount等。如果不使用内置函数我们可以运用Brian Kernighan算法。
Brian Kernighan 算法
该算法的意思是对于任何整数n,n&(n-1)该运算将n的二进制表示的最后一个1变成0,因此,对n重复该操作,直到n变成0,则操作次数即为n的二进制表示中1的个数。
代码
class Solution {
public:
vector<int> countBits(int n) {
vector<int> ans(n+1);
for(int i=0;i<=n;i++){
int count=0;
int num=i;
while(num>0){
num&=num-1;
count++;
}
ans[i]=count;
}
return ans;
}
};
优化
注意到在上面的算法中对每个整数都进行O(logn)的时间计算1的个数,但可以考虑使用动态规划的方法,如果存在0<=j<i,j的1个数已知,且i和j相比,i的二进制表示只多了一个1,则可以快速得到i的二进制中1的个数即ans[i]=ans[j]+1。所以寻找满足该条件的j值至关重要。对于随机一值x,如果y是2的整数次幂且是小于x的最大整数,则y的二进制表示必然只有一个1,且这个1在最高位。
例:x=10,10--->1010,比10小的最大整数且是2的整数次幂为8,8--->1000
ans[10]=ans[10-8]+1=ans[2]+1=2 (10-8的含义就是将最高位1去掉得到最高位后面几位中1的个数)
因为2的整数次幂最高位必然是1且后面全为0,由上面的Brian Kernighan 算法中可知,求一个2的整数次幂只要满足num&(num-1)==0即可。vector与普通数组的初始化 vector容器定义时如果程序中没有显式初始化,则容器会自动对所有元素进行初始化,例如vector数组所有元素会被自动初始化为0.
代码
class Solution {
public:
vector<int> countBits(int n) {
vector<int> ans(n+1);
int num=0;
for(int i=1;i<=n;i++){
if((i&(i-1))==0){
num=i;
}
ans[i]=ans[i-num]+1;
}
return ans;
}
};
注意vector容器定义时如果程序中没有显式初始化,则容器会自动对所有元素进行初始化0,因此当i为0和1的时候能够得到正确结果。