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)