1. 解析
题目大意,求解[0, n]内每个数字二进制表示的'1'个数的总和
2. 分析
比较容易想到的解法是,遍历[0, n],对于每个数字都表示成32位二进制,然后对每一位进行单独的统计。但题目明确要求O(n)的时间复杂度,上述方法不可行。如果仔细观察就可以发现。例如:n = 7
2 = , 3 = + 1, 4 = , 5 = + 1, 6 = + 2, 7 = + 3,即后面要求解的数一定可以拆分成,b是之前求解过的某个数。如果当前数是2的n次方,那么该数二进制表示'1'的个数总和就是1; 反之,拆分成两个数进行相加即可
class Solution {
public:
vector<int> countBits(int num) {
vector<int> res(num + 1, 0);
int cur = 1; //当前位
for (int i = 1; i <= num; ++i){
if (i > cur) cur = cur << 1;
if (i == cur){ //当前数是2的n次方,即对应的二进制位的1的总和为1
res[i] = 1;
continue;
}
res[i] = res[i - (cur >> 1)] + 1;
}
return res;
}
};