LeetCode - 338. Counting Bits

链接

338. Counting Bits

题意

给定一个整数num，计算从0到num之间的每个整数的二进制形式中1的个数。

思路

找规律：
0 -> [0]
1 -> [0, 1]
2 -> [0, 1, 1]
3 -> [0, 1, 1, 2]
4 -> [0, 1, 1, 2, 1]
...
7 -> [0, 1, 1, 2, 1, 2, 2, 3]
...
可知，每个数1的个数等于该数尾数%2加上该数除以2后1的个数，使用位运算更加高效

代码

Java:

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