Leetcode 338. 比特位计数
简单题
题目
给你一个整数 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 <= 1 0 5 10^5 105
进阶:
很容易就能实现时间复杂度为 O(n log n) 的解决方案,你可以在线性时间复杂度 O(n) 内用一趟扫描解决此问题吗?
你能不使用任何内置函数解决此问题吗?(如,C++ 中的 __builtin_popcount
)
解法
初级做法是将每个数计算其1的个数。使用countone
计算
class Solution:
def countBits(self, n: int) -> List[int]:
def countone(x):
res = 0
while x:
res += x%2
x = x//2
return res
result = []
for i in range(n+1): result.append(countone(i))
return result
也可以直接使用 x = x&(x-1) 消除最后一个1
class Solution:
def countBits(self, n: int) -> List[int]:
def countone(x):
res = 0
while x: x, res = x&(x-1), res+1
return res
result = []
for i in range(n+1): result.append(countone(i))
return result
- 时间复杂度 O(n logn) 每个数字最多遍历logn次,有n个数字
- 空间复杂度 O(n)
优化版本
比如:n=7
[0,1,2,3,4,5,6,7] num
[0,1,1,2,1,2,2,3] 1的个数
这种类型的题目需要从奇数,偶数的角度出发求解,同样利用前面的结果。
注意到奇数的1个数一定比前一个偶数个数多一个 只用在最后一位加1即可
偶数(num)1的个数和其一半(num/2) 1的个数是一样的 因为只是在其基础上左移一位
class Solution:
def countBits(self, n: int) -> List[int]:
result = [0]
for i in range(1,n+1):
if i%2: result.append(result[-1]+1) #奇数
else: result.append(result[i//2])
return result
- 时间复杂度 O(n)
- 空间复杂度 O(n)