其他分享
首页 > 其他分享> > 264. 丑数 II

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))

标签:heappop,丑数,heapq,10,II,heap,264,seen
来源: https://blog.csdn.net/qq_51174011/article/details/118738025