【leetcode】264. 丑数 II & 【剑指Offer】49.丑数

最新推荐文章于 2022-09-04 15:50:50 发布

原创最新推荐文章于 2022-09-04 15:50:50 发布 · 157 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 同时被 2 个专栏收录

116 篇文章

订阅专栏

剑指Offer

18 篇文章

订阅专栏

本文介绍了一种高效算法，用于查找第n个只包含质因数2,3,5的正整数（丑数）。通过动态规划方法，程序依次生成每个丑数，并最终返回第n个丑数。

题目：

编写一个程序，找出第 n 个丑数。

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

示例:

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

说明:

1 是丑数。
n 不超过1690。

代码：

class Solution:
    def nthUglyNumber(self, n: int) -> int:
        if n<= 0:
            return 0
        res = [0]*n
        res[0]=1
        nextindex = 1        
        p2 = 0
        p3 = 0
        p5 = 0
        while nextindex < n:
            minnum = min(res[p2]*2,res[p3]*3,res[p5]*5)
            res[nextindex] = minnum            
            while res[p2]*2 <= res[nextindex]:
                p2 +=1
            while res[p3]*3 <= res[nextindex]:
                p3 +=1
            while res[p5]*5 <= res[nextindex]:
                p5 +=1
            nextindex += 1
        return res[n-1]