LeetCode第338题——比特位计数（Java）

题目描述：

给你一个整数 n ，对于 0 <= i <= n 中的每个 i ，计算其二进制表示中 1 的个数 ，返回一个长度为 n + 1 的数组 ans 作为答案。

示例 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

提示：

• 0 <= n <= 105

思路：

• 通过观察可以发现奇数的二进制尾数都是1，偶数的二进制尾数都是0.奇数的1位数比前一个偶数的1位数多了一个尾数1

• 于是得到公式：

  ​ p[0] = 0; // 当n=0时

  ​ p[n] = p[n-1] + 1; // 当n为奇数时，n>0时

  ​ p[n] = p[n/2] // 当n为偶数时，n>0时

代码：

class Solution {
    public int[] countBits(int n) {
        int[] result = new int[n+1];
        int count = 0;
        if(n==0) return result;
        result[0] = 0;
        for(int i=1;i<n+1;i++){
            if(i%2==1)
                result[i] = result[i-1] + 1;
            else
                result[i] = result[i/2]; 
        }
        return result;
    }
}