力扣 338. 比特位计数

题目

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

示例

输入：n = 2
输出：[0,1,1]
解释：
0 --> 0
1 --> 1
2 --> 10

输入：n = 5
输出：[0,1,1,2,1,2]
解释：
0 --> 0
1 --> 1
2 --> 10
3 --> 11
4 --> 100
5 --> 101

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/counting-bits
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法1：库函数

Java实现

class Solution {
    public int[] countBits(int n) {
        int[] res = new int[n + 1];
        
        for (int i = 0; i <= n; i++) {
            res[i] = Integer.bitCount(i);
        }
        
        return res;
    }
}

方法2：模拟

Java实现

class Solution {
    public int[] countBits(int n) {
        int[] res = new int[n + 1];
        for (int i = 0; i <= n; i++) {
            res[i] = count(i);
        }
        return res;
    }

    public int count(int n) {
        int sum = 0;
        while (n != 0) {
            if (n % 2 == 1) sum++;
            n /= 2;
        }
        return sum;
    }
}

方法3：Brian Kernighan 算法

Java实现

class Solution {
    public int[] countBits(int n) {
        int[] res = new int[n + 1];
        for (int i = 0; i <= n; i++) {
            res[i] = countOnes(i);
        }
        return res;
    }

    public int countOnes(int x) {
        int res = 0;
        while (x > 0) {
            x = x & (x - 1);
            res++;
        }
        return res;
    }
}