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))
上一篇:1147 Heaps (30 分)--PAT甲级(判断堆),写完这题心态有点小崩


下一篇:懂你,更懂Rust系列之ownership