LeetCode 丑数问题II (三指针解决方法)

问题描述
编写一个程序,找出第 n 个丑数。

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

示例:

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

分析:丑数序列是从1开始乘以2,或者乘以3, 或者乘以5 的结果组成的序列
1x2=2, 1x3=3, 1x5=5, 2x2=4, 2x3=6, 2x5=10, 3x2=6,…
所以我们采用三指针方法, 记录从1来乘以了多少个2(或者3,或者5)
class Solution(object):

def nthUglyNumber(self, n):
    """
    :type n: int
    :rtype: int
    """
    res = [1]
    idx2 = 0
    idx3 = 0
    idx5 = 0
    for i in range(n-1):
        res.append(min(res[idx2]*2,res[idx3]*3,res[idx5]*5))
        if res[-1] == res[idx2]*2:
            idx2 += 1
        if res[-1] == res[idx3]*3:
            idx3 += 1
        if res[-1] == res[idx5]*5:
            idx5 += 1
    return res[-1]

你可能感兴趣的:(Review)