谈到这个题目,其实还有一个题目是该题目的基础。题目:求一个int型数转化为二进制后其中1的个数。首先我们需要知道怎么判断一个数是否是2的次方,即其转化为二进制后只有一个1,代码:
bool foo(int i) {
return !(i & (i - 1));
}
然后,我们计算int型转化为二进制后有多少个1的代码:
int countBit(int n) {
int count = 0;
while (n) {
n = n & (n -1);
count += 1;
}
return count;
}
回到我们的问题,如果在我们的问题中也用上述方法做,那么时间复杂度会比较高,达不到follow up中的要求。于是,在计算第i个数中转化为二进制后1的个数,需要通过i前面的某个数计算得到,不能是直接算出的。代码如下,聪明的你一定能够看懂。
我的代码:
class Solution {
public:
vector<int> countBits(int num) {
vector<int> res;
res.push_back(0);
int count = 0;
for (int i = 1; i <= num; ++ i) {
if (((i - 1) & i) == 0) {
count = 0;
} else {
count += 1;
}
res.push_back(1 + res[count]);
}
return res;
}
};
leetcode上大神们的代码,思路基本类似:
class Solution {
public:
vector<int> countBits(int num) {
vector<int> bits(num + 1, 0);
for (int i = 1; i <= num; ++ i) {
bits[i] = bits[i & (i - 1)] + 1;
}
return bits;
}
};
差距还很大,共勉!