题目
给定一个非负整数 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
直接法
遍历从 0 到 num ,针对每个数的情况做函数调用。
优化——动态规划法
发现递推规律,对于非负整数,只有两种情况,奇数和偶数
(1)奇数情况,二进制 1 的数字比它前面的偶数多 1 ;
(2)偶数情况,当偶数是 2 的次幂时,二进制位上 1 的个数为一个,例如分析 2、4、6、8、10 的情况,2(0010),4(0100),6(0110),8(1000) ,10(1010)
我们发现,非 2 的次幂偶数可以看作有限个 2 的次幂偶数的加合,并且加数的值并不重合。
右移思想
我们把非 2 次幂偶数右移一位,即可得到一个同样结果的出现过的数,只要让两者值相等即可,右移即四则运算中的除以 2 运算,理解为 2 乘以 一个奇数得到的结果,由于 2 (10),右移之后并不影响原来出现过的 1 的个数。
综上,偶数情况中为 2 的次幂时特殊情况,也符合右移特性,不必专门验证是否为 2 的次幂,8 / 2 = 4,4 / 2 = 2,等同于 2 的结果。
提交代码
直接法
class Solution:
def countBits(self, num: int) -> List[int]:
a = []
for i in range(0,num + 1):
a.append(self.count_num(i))
return a
def count_num(self,x):
if x == 0:
return 0
elif x == 1:
return 1
count = 0
while x & (x - 1) != 0:
count += 1
x = x & (x - 1)
count += 1
return count
优化——动态规划法
class Solution:
def countBits(self, num: int) -> List[int]:
dp=[0]*(num+1)
for i in range(1,num+1):
if(i%2==1):
dp[i]=dp[i-1]+1
else:
dp[i]=dp[i//2]
return dp
总结
- 偶数的二进制中 1 的个数,可通过右移,与得到的商的结果保持一致;
- 动态规划,避免在某种规律下相同结果的重复运算,直接找到规律索引位置对等的值即可。