标题
给你一个整数 n ,对于 0 <= i <= n 中的每个 i ,计算其二进制表示中 1 的个数 ,返回一个长度为 n + 1 的数组 ans 作为答案。
示例一
输入:n = 2
输出:[0,1,1]
解释:
0 --> 0
1 --> 1
2 --> 10
示例二
输入:n = 5
输出:[0,1,1,2,1,2]
解释:
0 --> 0
1 --> 1
2 --> 10
3 --> 11
4 --> 100
5 --> 101
思路
创建一个result数组,result[i]的含义是i所对应的二进制数的比特1个数。本文中,i的比特1个数可以由i/2的比特1个数轻易得到:
(1)若i能够整除i/2,说明i=i/2×2,则i只是在i/2的最右侧加了个比特0,而比特1的个数没有变化,则result[i]=result[i/2]+0;
(2)若i不能整除i/2,说明i=i/2×2+1,则i会在i/2的最右侧加个比特1,则result[i]=result[i/2]+1。
例如:
i=8时,i/2=4,由于i%(i/2)=0,所以result[8]=result[4]+0。(4的二进制100,8的二进制1000)
i=9时,i/2=4,由于i%(i/2)=1,所以result[9]=result[4]+1。(4的二进制100,9的二进制1001)
注意:0、1、2、3的result的值需要初始化时指定,动态规划的i须从4开始,因为当i=2或i=3时均可整除1,这是特殊情况。
C++代码
class Solution {
public:
vector<int> countBits(int n) {
vector<int> result;
//初始化
result.push_back(0);
if(n>=1)
result.push_back(1);
if(n>=2)
result.push_back(1);
if(n>=3)
result.push_back(2);
//动态规划
for(int i=4;i<=n;i++){
result.push_back(result[i/2]+i%(i/2));
}
return result;
}
};