给定一个非负整数 num。对于 0 ≤ i ≤ num 范围中的每个数字 i ,计算其二进制数中的 1 的数目并将它们作为数组返回。
示例 1:
输入: 2
输出: [0,1,1]
示例 2:
输入: 5
输出: [0,1,1,2,1,2]
思路:
这道题用动态规划来解,我们举一个例子:
我们可以很容易的得知,凡为2的幂的数写成二进制以后1的数目都为1。比如2:10,4:100,8:1000,16:10000 等等
对于不为2的幂的数,如7=4+3,我们可以将它分为一个2的当前最高次幂4和一个非2的幂3,那么7的二进制数含1的数量等于4的二进制数含1的个数 1+ 3的二进制数含1的数量 2 =3。
例如15,我们同样可以将其分解为8+7。那么15的二进制数含1的个数=1+(7的二进制数含1的个数),我们之前已经计算过7了,直接拿来用就可以了。所以15的二进制数含1的个数=4。
可以自己把从1到16的二进制数都写一遍。
从0遍历输入的数字,我们用一个数组result来保存当前数字。用一个整型变量flag来保存目前为止最大的2的幂。
判断一个数是否为2的幂((i&(i-1))==0),如果i为2的幂,将flag=i,然后result[i]=1。如果i不为2的幂,那么result[i]=result[i-flag]+1;
时间复杂度O(n)
class Solution {
public int[] countBits(int num) {
int[] results = new int[num+1];
results[0]=0;
int flag=0;
for(int i=1;i<=num;i++)
{
if((i & (i - 1)) == 0)
{
results[i]=1;
flag = i;
}
else
{
results[i] = results[i-flag]+1;
}
}
return results;
}
}