剑指 Offer II 003. 前 n 个数字二进制中 1 的个数

题目

链接：https://leetcode.cn/problems/w3tCBm/

给定一个非负整数 n ，请计算 0 到 n 之间的每个数字的二进制表示中 1 的个数，并输出一个数组。

思路

对每个数字，逐位判断是否存在1。当一个数为奇数时，二进制的最末位是1。所以我们可以根据判断当前的数字是否为奇数，来判断末位是否为1。之后通过右移移除最后一位，然后在判断右移后的数字是否为奇数。

这个方法固然可行，但是进阶里提出了更高的要求，是否可以实现 O(n)内用一趟扫描做到。

那么我们是不是可以找一找数字之间的规律，看是否可以利用数组中已存在的值，计算出当前的值？

解法一：对数字逐位判断

代码

public int[] countBits(int n) {
    int[] result = new int[n + 1];
    for (int i = 0; i <= n; i++) {
        int j = i;
        int current = 0;
        while (j > 0) {
            current += j % 2;
            j = j >> 1;
        }
        result[i] = current;
    }

    return result;
}

解法二：奇数偶数的逻辑

逻辑

很简单的几个逻辑：

1. 二进制里，偶数的最后一位是0，奇数的最后一位是1。
2. 由第一点可知，一个奇数的值，等于比他小1的那个偶数的最后一位置为1。比如数字6，对应的二进制是110。数字7，对应的二进制是111。
3. 一个偶数，对他进行右移1位，会移除掉最后一位的0。

所以，对于本题来说：

• 对于奇数n，他二进制里的1的个数就是n-1的二进制中1的个数加1。
• 对于偶数n，那么二进制里的1的个数与n/2的二进制中1的个数相等。
• 而n-1和n/2的二进制中1的个数，已经保存在前面的数组里了。

代码

public int[] countBits(int n) {
    int[] result = new int[n + 1];
    result[0] = 0;
    for (int i = 1; i <= n; i++) {
        result[i] = i % 2 == 1 ? result[i - 1] + 1 : result[i / 2];
    }

    return result;
}

扩展思考

1. 遇到和二进制相关的问题，要多试试二进制的位运算。
2. 遇到时间复杂度为O(n)的题，要多想想，是否可以借助已经存储的数据，来降低时间复杂度。