LeetCode 每日一题338. 比特位计数

338. 比特位计数

给定一个非负整数 num。对于 0 ≤ i ≤ num 范围中的每个数字 i ，计算其二进制数中的 1 的数目并将它们作为数组返回。

示例 1:

输入: 2
输出: [0,1,1]

示例 2:

输入: 5
输出: [0,1,1,2,1,2]

进阶:

• 给出时间复杂度为O(n*sizeof(integer)) 的解答非常容易。但你可以在线性时间 O(n) 内用一趟扫描做到吗？
• 要求算法的空间复杂度为 O(n)。
• 你能进一步完善解法吗？要求在C++或任何其他语言中不使用任何内置函数（如 C++ 中的 __builtin_popcount）来执行此操作。

方法一：动态规划

在介绍「动态规划」之前先介绍两种其他解法。

1. 调用系统函数 Integer.bigCount(int i) 计算 i 的二进制位有多少个一。函数实现参考 这篇博客。

public int[] countBits(int num) {
    int[] ans = new int[num + 1];
    for (int i = 0; i <= num; i++) {
        ans[i] = Integer.bitCount(i);
    }
    return ans;
}

2. 手写计算 bitCount(int x) 函数。核心代码是 x &= (x - 1)，效果是去掉 x 最低位的 1，通过这种方式把 x 逐渐缩小到 0。

public int[] countBits(int num) {
    int[] bits = new int[num + 1];
    for (int i = 0; i <= num; i++) {
        bits[i] = bigCount(i);
    }
    return bits;
}
public int bigCount(int x) {
    int count = 0;
    while (x > 0) {
        x &= (x - 1);
        count += 1;
    }
    return count;
}

动态规划写法

• 定义 bits 数组，bits[i] 表示 数字 i 二进制位有多少个 1。
• 由于 x &= (x - 1) 的效果是去掉最低位的 1。假设我们当前求 bits[x]， 我们只用找到 x 去除最低位 1 的那个数即 x & (x - 1) ，有：bits[x] = bits[x & (x - 1)] + 1。

参考代码

public int[] countBits(int num) {
    int[] bits = new int[num + 1];
    for (int i = 1; i <= num; i++) {
        bits[i] = bits[i & (i - 1)] + 1;
    }
    return bits;
}

执行结果