LeetCode338. 比特位计数

题目

给定一个非负整数 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）来执行此操作。

分析

从0到n每个数字的二进制数字中1的个数，可以每个数字都算一遍，算的方法也比较多，可以用位运算来算同1做&与运算，也可以用比较慢的方法转成字符串然后挨个遍历字符串……
上面说的方法的时间复杂度都降不到O(n)，如果想要一遍扫描就得到结果的话，还是要借助之前的结果，然后自己在纸上写了一下。

那我们可以观察到，逢2进位，10是1的下一个数字进了位，进位就会多一个1，所以每进一位都相当于之前的数字加上1。
那，
2-3的1的个数 = 0-1的1的数字个数+1
4-7的1的个数 = 0-3的1的数字个数+1
8-15的1的个数 = 0-7的1的数字个数+1
……

nums[i] = 1 + nums[ i - 2^log2(i)]

代码

class Solution {
    public int[] countBits(int num) {
        int[] nums = new int[num+1];
        if (num == 0) return nums;
        nums[1] = 1;
        int temp = 0;
        int subNum = 0;

        for (int i = 2; i <= num; i++) {
            if (i == 2 << temp){
                subNum = i;
                temp ++;
            }
            nums[i] = nums[i-subNum] + 1;
        }

        return nums;
    }
}