本题的题意是输入一个非负整数,求从0到这个非负整数之间每一个整数的二进制表示中“1”的个数,并且题目要求算法的时间和空间复杂度均为O(n)。
我采用动态规划算法来解决此题。我们不难发现,每一个2的整数幂的二进制表示中都只有一个“1”,并且每一个非2的整数幂的二进制表示中“1”的个数都可以这样计算:该数减去比它小的最大的2的整数幂,那么得到的数的二进制表示中“1”的个数再加上1就是该数的二进制表示中“1”的个数。(0的二进制表示中“1”的个数为0。)也就是说,我们只要在从0到num的遍历过程中,存储已经计算出的每个数的二进制表示中“1”的个数,就可以直接利用之前算出的值计算后面的值。这样的话,最后返回计算出的数组即可,且只需遍历一次,空间复杂度也是O(n),满足要求。
class Solution {
public:
vector<int> countBits(int num) {
vector<int> result;
int current = 0;
int future = 1;
result.push_back(0);
for (int i = 1; i <= num; ++i) {
if (i < future) {
result.push_back(result[i - current] + 1);
} else if (i == future) {
result.push_back(1);
current = future;
future *= 2;
}
}
return result;
}
};