leetcode笔记之[338. 比特位计数]

103 阅读1分钟

Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情

一、题目描述:

338. 比特位计数 - 力扣(LeetCode) (leetcode-cn.com)

给你一个整数 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 )

二、思路分析:

转为bin统计1的个数,使用bin(num[i])[2:]得到二进制表示

使用str.count("1")统计1的个数。

三、AC 代码:

class Solution:
    def countBits(self, num: int) -> List[int]:
        #简单思路:转为bin统计1的个数
        num = list(range(num+1))
        size = len(num)
        for i in range(size):
            num[i] = bin(num[i])[2:]
            num[i] = num[i].count("1")
        return num

四、总结:

只出题解还是很容易的,难就难在进阶的要求时间复杂度低后的解法。

范文参考

官方题解:最高有效位(将二进制比如 10001看成两部) - 比特位计数 - 力扣(LeetCode) (leetcode-cn.com)

执行用时: 1 ms , 在所有 Java 提交中击败了 99.92% 的用户 - 比特位计数 - 力扣(LeetCode) (leetcode-cn.com)