题目描述:
给你一个整数 n ,对于 0 <= i <= n 中的每个 i ,计算其二进制表示中 1 的个数 ,返回一个长度为 n + 1 的数组 ans 作为答案。
示例:
输入:n = 5
输出:[0,1,1,2,1,2]
解释:
0 --> 0
1 --> 1
2 --> 10
3 --> 11
4 --> 100
5 --> 101
算法思想
位运算,奇偶表达规律。
对于正整数n来说,2*n的二进制表示即将n左移一位,0的个数+1,1的个数未变。
所以可以得知,任何偶数i的二进制表达中1的个数为res[i] = res[i/2]; (i%2 == 0)
而任何奇数 x 可以表示为偶数j+1或k-1的形式(因为题目数字从0开始,所以取偶数j+1),而这个偶数j中1的个数可以表示为res[j]=res[j/2],且末尾一定是0,那么这个奇数x=j+1,所以res[x] = res[j]+1=res[j/2]+1。res[0] = 0, res[1] = 1,依次向后递推即可。
代码实现
class Solution {
public:
vector<int> countBits(int n) {
vector<int> res(n+1);
for(int i = 0; i <= n; i++){
if(i % 2 == 0) //偶数
res[i] = res[i/2];
else //奇数
res[i] = res[i/2] + 1;
}
return res;
}
};