给定一个非负整数 num。对于 0 ≤ i ≤ num 范围中的每个数字 i ,计算其二进制数中的 1 的数目并将它们作为数组返回。
输入: 2 输出: [0,1,1]
输入: 5 输出: [0,1,1,2,1,2]
暴力破解:
class Solution {
private:
int countNum(int num) {
int res = 0;
while(num >= 1) {
if(num % 2 == 1) res++;
num = num / 2;
}
return res;
}
public:
vector<int> countBits(int n) {
vector<int> bits(n+1,0);
for(int i = 0; i < n+1; i++) {
bits[i] = countNum(i);
}
return bits;
}
};
位运算的改进: 这里已经写过博客