开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第11天,点击查看活动详情
一、题目描述:
给你一个整数
n,对于0 <= i <= n中的每个i,计算其二进制表示中1的个数 ,返回一个长度为n + 1的数组ans作为答案。
示例 1:
输入:n = 2
输出:[0,1,1]
解释:
0 --> 0
1 --> 1
2 --> 10
示例 2:
输入:n = 5
输出:[0,1,1,2,1,2]
解释:
0 --> 0
1 --> 1
2 --> 10
3 --> 11
4 --> 100
5 --> 101
提示:
- 0 <= n <= 10^5
进阶:
- 很容易就能实现时间复杂度为 O(n log n) 的解决方案,你可以在线性时间复杂度 O(n) 内用一趟扫描解决此问题吗?
- 你能不使用任何内置函数解决此问题吗?(如,C++ 中的 __builtin_popcount )
二、思路分析:
看到这种题目,首先想到的就是迭代法,就是直接两次遍历进行求解:
- 外层循环,用index来遍历0-n
- 内层循环,用来记录index这个数里面有多少个1
三、AC 代码:
class Solution:
def countBits(self, n: int) -> List[int]:
res=[0 for _ in range(n+1)]
for i in range(1,n+1):
n=i
count=0
while n!=0:
count+=n&1
n>>=1
res[i]=count
return res
四、总结:
其实该题目也是可以使用动态规划来求解的,我们可以根据二进制的串的性质推算出其递推公式:
当index的二进制表示的最后一位是1时,那么dp[index]一定比dp[index-1]多1
当index的二进制表示的最后一位是0时,那么dp[index]一定和dp[index>>1]的数值一样
五、参考:
学会这道题,再AC两题!C++, 布赖恩·克尼根算法 + 动态规划。 - 比特位计数 - 力扣(LeetCode)
【仗剑骑士】C++, 两种方法实现O(n)时间复杂度, 统计规律+动态规划+338. 比特位计数 - 比特位计数 - 力扣(LeetCode)