[leetcode] 264. 丑数 II python解法

给你一个整数 n ,请你找出并返回第 n 个 丑数 。

丑数 就是只包含质因数 2、3 和/或 5 的正整数。

示例 1:

输入:n = 10
输出:12
解释:[1, 2, 3, 4, 5, 6, 8, 9, 10, 12] 是由前 10 个丑数组成的序列。
示例 2:

输入:n = 1
输出:1
解释:1 通常被视为丑数。

思路:动态规划,用还没乘过 2 的最小丑数乘以 2;用还没乘过 3 的最小丑数乘以 3;用还没乘过 5 的最小丑数乘以 5。然后在得到的数字中取最小,就是新的丑数。

class Solution(object):
    def nthUglyNumber(self, n):
        if n < 0:
            return 0
        dp = [1] * n
        i2, i3, i5 = 0, 0, 0
        for i in range(1, n):
            dp[i] = min(2 * dp[i2], 3 * dp[i3], 5 * dp[i5])
            if dp[i] == 2 * dp[i2]: i2 += 1
            if dp[i] == 3 * dp[i3]: i3 += 1
            if dp[i] == 5 * dp[i5]: i5 += 1
        return dp[n - 1]

用3个if:去重

你可能感兴趣的:(leetcode)