利用上一节的n&(n-1)的性质:
动态规划 + 最高有效位
利用已有的计数结果来生成新的计数结果。以二进制形式检查 [0, 3]的范围:
可以看出, 2 和 3 的二进制形式可以通过给 0 和 1 的二进制形式在前面加上 1 来得到。类似的,我们可以使用 [0, 3]作为蓝本来得到 [4, 7]。
public class Solution {
public int[] countBits(int num) {
int[] ans = new int[num + 1];
int i = 0, b = 1;
// [0, b) is calculated
while (b <= num) {
// generate [b, 2b) or [b, num) from [0, b)
while(i < b && i + b <= num){
ans[i + b] = ans[i] + 1;
++i;
}
i = 0; // reset i
b <<= 1; // b = 2b
}
return ans;
}
}
动态规划 + 最低有效位
通过最低有效位来获得状态转移函数。观察x 和 x’ = x / 2的关系,可以发现 x’与 x只有一位不同,这是因为x’可以看做 x 移除最低有效位的结果。
public class Solution {
public int[] countBits(int num) {
int[] ans = new int[num + 1];
for (int i = 1; i <= num; ++i)
ans[i] = ans[i >> 1] + (i & 1); // x / 2 is x >> 1 and x % 2 is x & 1
return ans;
}
}
动态规划 + 最后设置位
最后设置位是从右到左第一个为1的位。使用 x &= x - 1 将该位设置为0,就可以得到以下状态转移函数:
X&(X-1)的结果指的是将X的最后一位为1的bit变为1,那么X肯定会比该数多1位为1。
public class Solution {
public int[] countBits(int num) {
int[] ans = new int[num + 1];
for (int i = 1; i <= num; ++i)
ans[i] = ans[i & (i - 1)] + 1;
return ans;
}
}