LeetCode 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
)来执行此操作。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/counting-bits
思路解析
题目中要求时间复杂度为 (O(n)),但是 (O(1)) 的时间复杂度下无法计算一个数字的 1bit
数,因此考虑动态规划。
例如:9(1001) 比 1(1) 的 1bit
数多1,因为 9 仅在最高位上多出了一个1,因此我们需要考虑记录一个最高位,一旦知道了当前(i
)的最高位为 highBit
, 那么可得状态转移方程:
bit[i] = bit[i - highBit] + 1
问题在于如何在 (O(1)) 的时间复杂度下记录当前的最高位?
直接给出结论:
-
(i & (i - 1)) == 0
,则 i 为 (2^n)。
代码实现
class Solution {
public:
vector<int> countBits(int num) {
vector<int> bit(num + 1);
int highBits = 0;
for(int i = 1; i <= num; i++) {
if((i & (i - 1)) == 0)
highBits = i;
bit[i] = bit[i - highBits] + 1;
}
return bit;
}
};