[leetcode] 264. 丑数 II python解法

最新推荐文章于 2022-12-30 16:12:47 发布

喝牛奶的草履虫

最新推荐文章于 2022-12-30 16:12:47 发布

阅读量241

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/milk_paramecium/article/details/118733429

版权

leetcode 专栏收录该内容

77 篇文章 1 订阅

订阅专栏

给你一个整数 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]