Python3 LintCode代码 4.丑数 II

设计一个算法,找出只含素因子235 的第 n 小的数。

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

丑数的因子只含有2,3,5。我们默认将1作为第一个丑数。在此算法中,我们利用这一定义,2,3,5乘以2,3,5的倍数也一定是丑数,于是,将每轮取得的最小的丑数加入列表,直到取到我们所需要的丑数。

class Solution:
    """
    @param n: An integer
    @return: the nth prime number as description.
    """
    def nthUglyNumber(self, n):
        # write your code here
        L = [1]
        i2,i3,i5 = 0,0,0
        for k in range(n):
        	n2,n3,n5 = L[i2]*2,L[i3]*3,L[i5]*5
        	m = min(n2,n3,n5)
        	L.append(m)
        	i2 += (m == n2)
        	i3 += (m == n3)
        	i5 += (m == n5)
        return L[n-1]
test = Solution()
print(test.nthUglyNumber(9))

猜你喜欢

转载自blog.csdn.net/kevin_pei/article/details/79489654
今日推荐