264. 丑数 II

264. 丑数 II

给你一个整数 n ,请你找出并返回第 n 个 丑数 。

丑数 就是只包含质因数 2、3 和/或 5 的正整数。

示例 1:
输入:n = 10
输出:12
解释:[1, 2, 3, 4, 5, 6, 8, 9, 10, 12] 是由前 10 个丑数组成的序列。
示例 2:
输入:n = 1
输出:1
解释:1 通常被视为丑数。
提示:
1 <= n <= 1690

解题:最小堆

首先,第一个反应是暴力解,先用分解质因数的方法写一个 判断一个数是否为丑数的函数 ,最后便利数组,找出丑数。

但是这样写的时间复杂度特别大。因为随着数的增大,丑数的分布可能会变得稀疏,其次,对于每一个数,判断它是否为丑数的成本增长飞快。

于是,我们开始思考,解决问题的方法。第一步,就是找出丑数的变幻法制。那么最简单的一个,就是对于丑数x,其2x,3x,5x也是丑数。根据这个递推法制,再结合“第n个丑数”的条件,我们考虑用最小堆的数据结构,存储递推得到的丑数,并通过heappop和heapify的方式来完成递推。

factors = [2, 3, 5]
seen = {1}
heap = [1]

for i in range(n - 1):
    curr = heapq.heappop(heap)
    for factor in factors:
        if (nxt := curr * factor) not in seen:
            seen.add(nxt)
            heapq.heappush(heap, nxt)

print(heapq.heappop(heap))
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Grayson Zhang

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值