前 n 个数字二进制中 1 的个数
题目:
给定一个非负整数 n ,请计算 0 到 n 之间的每个数字的二进制表示中 1 的个数,并输出一个数组。
示例 1:
输入: n = 2
输出: [0,1,1]
解释:
0 --> 0
1 --> 1
2 --> 10
示例 2:
输入: n = 5
输出: [0,1,1,2,1,2]
解释:
0 --> 0
1 --> 1
2 --> 10
3 --> 11
4 --> 100
5 --> 101
class Solution {
public int[] countBits(int n) {
List<Integer> list = new ArrayList();
for(int i = 0; i <= n; i++) {
list.add(getOne(i));
}
int[] ans = new int[list.size()];
for(int i = 0; i < list.size(); i++) {
ans[i] = list.get(i);
}
return ans;
}
private int getOne(int n) {
int sum = 0;
while(n > 0) {
sum += n & 1;
n >>= 1;
}
return sum;
}
}
/**
对于所有的数字,只有奇数和偶数两种:
奇数:二进制表示中,奇数一定比前面那个偶数多一个 1,因为多的就是最低位的 1。
偶数:二进制表示中,偶数中 1 的个数一定和除以 2 之后的那个数一样多。因为最低位是 0,除以 2 就是右移一位,也就是把那个 0 抹掉而已,所以 1 的个数是不变的。
综上可以得出:
当 i 为奇数时 dp[i] = dp[i - 1] + 1
当 i 为偶数时 dp[i] = dp[i / 2]
又因为偶数中 1 的个数一定和除以 2 之后的那个数一样多,所以:
当 i 为奇数时 dp[i] = dp[i - 1] + 1 -> dp[i] = dp[i / 2] + 1
化简以后可以得到:
dp[i] = dp[i / 2] + (i & 1)
**/
class Solution {
public int[] countBits(int n) {
int[] bits = new int[n + 1];
for (int i = 1; i <= n; i++) {
bits[i] = bits[i >> 1] + (i & 1);
}
return bits;
}
}