264. 丑数 II

F=k·Δx

于 2021-07-14 16:07:06 发布

阅读量75

点赞数

分类专栏： Python 文章标签：动态规划丑数质因数指针动态规划数组

本文链接：https://blog.csdn.net/Outsider_ahhdd/article/details/118730125

版权

Python 专栏收录该内容

158 篇文章 3 订阅

订阅专栏

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

提示：

1 <= n <= 1690

思路

官方解法的动态规划
主要难点是想到用三个指针去求解，每次放入结果集中的是各自指针对应的dp值乘以其系数，当前指针乘以系数后的值一定是被放入dp后，才会移动当前指针，保证了dp值从小到大，达到第n位

代码

class Solution:
    def nthUglyNumber(self, n: int) -> int:
        dp = [0] *(n + 1)
        dp[1] = 1
        p2 = p3 = p5 = 1
        for i in range(2,n+1):
            a = dp[p2]*2
            b = dp[p3]*3
            c = dp[p5]*5
            dp[i] = min(a,b,c)
            if dp[i] == a:
                p2 += 1
            if dp[i] == b:
                p3 += 1
            if dp[i] == c:
                p5 += 1       
        return dp[n]