Leetcode388 比特位计数题解

题目描述

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

解法一：利用位运算的性质

按位与运算（&）的一个性质是：对于任意整数 x，令 x=x&(x−1)，该运算将 x 的二进制表示的最后一个 1 变成 0。因此，对 x 重复该操作，直到 x 变成 0，则操作次数即为 x 的二进制数中的 1 的数目。

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

时间复杂度：$O(k\times \text{num})$，其中 $k$ 是 $\text{int}$ 型的二进制位数，$k=32$。需要对从 $0$ 到 $\text{num}$ 的每个数使用 $O(k)$ 的时间计算
空间复杂度：$O(1)$ 除了返回的数组以外，空间复杂度为常数。

这样的时间复杂度显然无法让我们满意。

解法二：观察分析法

奇数：二进制表示中，奇数一定比前面那个偶数多一个 1，因为多的就是最低位的 1。
偶数：二进制表示中，偶数中 1 的个数一定和除以 2 之后的那个数一样多。因为最低位是 0，除以 2 就是右移一位，也就是把那个 0 抹掉而已，所以 1 的个数是不变的。
为了增加运算速度，我们采用位运算来判断奇数偶数。

class Solution {
    public int[] countBits(int num) {
      int[] result = new int[num + 1];
      for(int i = 1; i <= num; i++){
        if ((i & 1) == 0){
          result[i] = result[i >> 1];
        }else {
          result[i] = result[i - 1] + 1;
        }
      }
      return result;
    }
}

但是判断中又稍显复杂，再观察，其实能发现，其实奇数，也是向左移一位加上1，于是

class Solution {
    public int[] countBits(int num) {
      int[] result = new int[num + 1];
      for(int i = 1; i <= num; i++){
          result[i] = result[i >> 1] + (i & 1);
      }
      return result;
    }
}

时间复杂度：$O(\text{num})$，对从 $0$ 到 $\text{num}$ 的每个数只需要使用 $O(1)$ 的时间计算
空间复杂度：$O(1)$ 除了返回的数组以外，空间复杂度为常数。