前 n 个数字二进制中 1 的个数

前 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;
    }
}