给定一个非负整数 n
,请计算 0
到 n
之间的每个数字的二进制表示中 1 的个数,并输出一个数组。
示例 2:
输入: n = 5
输出: [0,1,1,2,1,2]
解释:
0 --> 0
1 --> 1
2 --> 10
3 --> 11
4 --> 100
5 --> 101要求用O(n)复杂度
class Solution {
public int[] countBits(int n) {
int[] dp = new int[n+1];
// 转移方程:
// dp[i] = dp[i/2] (i是偶数) 因为2倍相当于左移一位,低位多出的是0对1的个数无影响
// dp[i] = dp[i-1]+1 (i是奇数) 因为相邻的偶数变基数区别在于最低为0-->1,所以只会多1个1
// base case : dp[0] = 0
// i/2 --> i>>1
// i是偶数则加0,i是基数则加1 --> +(i&1),因为奇偶由二进制最低位决定,
// 且i为基数时i-1是偶数,dp[i-1] = dp[i>>1]
for (int i=1; i<=n; i++) {
dp[i] = dp[i>>1] + (i&1);
}
return dp;
}
}