给定一数字,分别统计从零到该数字之间所有数字二进制中 1 的个数
从hint中可以看出此题可用DP解,进一步观察可发现只要是2的幂,1的个数均为1,而两个相邻幂之间的数字与前一个幂数有关
判断一个数是否为2的幂可用位运算:
if((n & (n-1)) == 0) //该数是2的幂
代码如下:
class Solution {
public:
vector<int> countBits(int num) {
vector<int> dp(num+1);
int curr;
dp[0] = 0;
if(num>=1) //注意
dp[1] = 1;
for(int i=2; i<=num; ++i)
{
if((i&(i-1)) == 0)
{
curr = i;
dp[i] = 1;
}
if((i-curr) > 0)
dp[i] = 1 + dp[i-curr];
}
return dp;
}
};
需要注意的是,当输入为零时,开辟了dp[1],但不应存储数字,所以应把该情况排除