leetcode338. 比特位计数

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

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

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

我的想法很简单，就是用一个数组保存值，然后每次+1时，进行计算，我想了下复杂度，也是o(n)级别的。

public int[] countBits(int num) {
    int[] ans = new int[num + 1];
    int[] res = new int[32];
    int an = 0;
    for (int i = 1; i <= num; i++) {
        int add = 1;
        for (int j = 0; j < 32; j++) {
            if (add == 0) {
                break;
            }
            an -= res[j];
            res[j] += add;
            if (res[j] == 2) {
                res[j] = 0;
            } else {
                add = 0;
            }
            an += res[j];
        }
        ans[i] = an;
    }
    return ans;
}

但是我看到时间最短的方法时，让我很惊讶，看了能理解，但是自己应该想不到。

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

分析下为什么成立。
1、当i%2=1时，这个比较好理解，i&(i-1)=i-1,res[i]=res[i-1]+1,差距就在最后一位。
2、当i%2=0时，(i-1)尾部有连续k个1，倒数第(k+1)位是0,自然，i的尾部有连续k个0，倒数第(k+1)位是1，res[i] = res[i & (i-1)] + 1;也能理解