python--lintcode4. 丑数 II

描述

设计一个算法，找出只含素因子2，3，5 的第 n 小的数。

符合条件的数如：1, 2, 3, 4, 5, 6, 8, 9, 10, 12...

我们可以认为1也是一个丑数

您在真实的面试中是否遇到过这个题？  

样例

如果n = 9， 返回 10

挑战

要求时间复杂度为O(nlogn)或者O(n)

这一题的思路直接看代码吧，就几行：

class Solution:
    """
    @param n: An integer
    @return: the nth prime number as description.
    """
    def nthUglyNumber(self, n):
        # write your code here
        ugly=set()
        ugly.add(1)
        minNumber=1
        for i in range(n):
            minNumber=min(ugly)
            ugly.remove(minNumber)
            ugly.add(minNumber*2)
            ugly.add(minNumber*3)
            ugly.add(minNumber*5)
        return minNumber




s=Solution()
print(s.nthUglyNumber(10))