【NO.82】LeetCode HOT 100—338. 比特位计数

338. 比特位计数

给你一个整数 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

进阶：

很容易就能实现时间复杂度为 O(n log n) 的解决方案，你可以在线性时间复杂度 O(n) 内用一趟扫描解决此问题吗？
你能不使用任何内置函数解决此问题吗？（如，C++ 中的 __builtin_popcount ）

解题

方法一：与运算

// 时间复杂度：O(nlog⁡n) 空间复杂度：O(1)
class Solution {
    public int[] countBits(int n) {
        int[] ans = new int[n+1];
        for (int i = 0; i <= n; i++) {
            ans[i] = getCount(i);
        }
        return ans;
    }

    // 获取x的二进制表示中 1 的个数
    private int getCount(int x) {
        int count = 0;
        while (x != 0) {
            // 与运算，如果相对应位都是1，则结果为1，否则为0
            // 而1的二进制，只有最右边为1，
            // 所以我们逐渐右移x的二进制数，让其最右边与1做与运算，得到的结果相加就是x的二进制表示中 1 的个数
            count += x & 1;
            x >>= 1;
        }

        return count;
    }
}

方法二：动态规划——最低有效位

对于正整数 x，将其二进制表示右移一位，等价于将其二进制表示的最低位去掉，得到的数是 ⌊x/2⌋。如果 bits[⌊x/2⌋]的值已知，则可以得到 bits[x] 的值：

• 如果 x 是偶数，则 bits[x]=bits[⌊x/2⌋]

• 如果 x 是奇数，则 bits[x]=bits[⌊x/2⌋]+1

上述两种情况可以合并成：bits[x] 的值等于 bits[⌊x/2⌋]的值加上 x 除以 2 的余数。

由于 ⌊x/2⌋可以通过 x>>1 得到，x 除以 2 的余数可以通过 x & 1得到，因此有：bits[x]=bits[x>>1]+(x & 1)

遍历从 1 到 n 的每个正整数 i，计算 bits 的值。最终得到的数组 bits即为答案。

//时间复杂度：O(n)
//空间复杂度：O(1)
class Solution {
    public int[] countBits(int n) {
        int[] ans = new int[n+1];
        for (int i = 0; i <= n; i++) {
            ans[i] = ans[i >> 1] + (i & 1);
        }
        return ans;
    }

}