264.丑数II

最新推荐文章于 2021-06-24 10:00:13 发布

Z0o1010

最新推荐文章于 2021-06-24 10:00:13 发布

阅读量80

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/weixin_48732879/article/details/115598843

版权

LeetCode 专栏收录该内容

258 篇文章 0 订阅

订阅专栏

给你一个整数 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 通常被视为丑数。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/ugly-number-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

# 生成n个丑数并返回最后一个即可
class Solution(object):
    def nthUglyNumber(self, n):
        if n < 0:
            return 0

        res = [1] * n
        ind2, ind3, ind5 = 0, 0, 0  # 初始化还没乘过2、3、5的丑数位置

        for i in range(1, n):
            res[i] = min(2 * res[ind2], 3 * res[ind3], 5 * res[ind5])  # 用还没乘过 2 的最小丑数乘以 2；用还没乘过 3 的最小丑数乘以 3；用还没乘过 5 的最小丑数乘以 5。然后在得到的数字中取最小，就是新的丑数。
            
            if res[i] == 2 * res[ind2]:
                ind2 += 1
            if res[i] == 3 * res[ind3]:
                ind3 += 1
            if res[i] == 5 * res[ind5]:
                ind5 += 1

        return res[-1]