338. 比特位计数
给定一个非负整数 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)来执行此操作。
方法一:动态规划
在介绍「动态规划」之前先介绍两种其他解法。
- 调用系统函数
Integer.bigCount(int i)
计算i
的二进制位有多少个一。函数实现参考 这篇博客。
public int[] countBits(int num) {
int[] ans = new int[num + 1];
for (int i = 0; i <= num; i++) {
ans[i] = Integer.bitCount(i);
}
return ans;
}
- 手写计算 bitCount(int x) 函数。核心代码是
x &= (x - 1)
,效果是去掉x
最低位的1
,通过这种方式把x
逐渐缩小到0
。
public int[] countBits(int num) {
int[] bits = new int[num + 1];
for (int i = 0; i <= num; i++) {
bits[i] = bigCount(i);
}
return bits;
}
public int bigCount(int x) {
int count = 0;
while (x > 0) {
x &= (x - 1);
count += 1;
}
return count;
}
动态规划写法
- 定义 bits 数组,
bits[i]
表示数字 i
二进制位有多少个 1。 - 由于
x &= (x - 1)
的效果是去掉最低位的1
。假设我们当前求bits[x]
, 我们只用找到x
去除最低位1
的那个数即x & (x - 1)
,有:bits[x] = bits[x & (x - 1)] + 1
。
参考代码
public int[] countBits(int num) {
int[] bits = new int[num + 1];
for (int i = 1; i <= num; i++) {
bits[i] = bits[i & (i - 1)] + 1;
}
return bits;
}