丑数2 python_LeetCode - Python - 264. 丑数 II - 期权论坛

编写一个程序，找出第 n 个丑数。

丑数就是只包含质因数 2, 3, 5 的正整数。

示例:

输入: n = 10

输出: 12

解释: 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 是前 10 个丑数。

说明:

1 是丑数。

n 不超过1690。

来源：力扣(LeetCode)

链接：https://leetcode-cn.com/problems/ugly-number-ii

著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

第一种思路：

麻瓜解法，从1开始往后每个数都判断一下是不是丑数，直到找到第n个丑数位置。

太麻瓜了，没有实现。

第二种思路：

思考一下可以发现，除第一个丑数1外，每个丑数都可以由之前的丑数 * 2 或3 或5 得到，

比如 1 * 2 = 2， 1 * 3 = 3， 1 * 5 = 5， 2 * 2 = 4， 2 * 3 = 6， 2 * 5 = 10。。。。。

所以找到所有的丑数很简单，挨个乘 2 或3 或5即可，

下一步要确定如何按顺序找到丑数，

显然，丑数的顺序遵循从小到大的顺序，因此可以理解为动态找最小值，

所以用最小堆就可以实现，

注意，因为会出现 2 * 3 = 6 和 3 * 2 = 6的情况，所以需要开一个set避免重复。

class Solution(object):

def nthUglyNumber(self, n):

"""

:type n: int

:rtype: int

"""

from heapq import *

l = [1]

heapify(l)

cnt = 1

used = set([1])

while cnt < n:

cur = heappop(l)

if cur * 2 not in used:

heappush(l, cur * 2)

used.add(cur * 2)

if cur * 3 not in used:

heappush(l, cur * 3)

used.add(cur * 3)

if cur * 5 not in used:

used.add(cur * 5)

heappush(l, cur * 5)

cnt += 1

return heappop(l)