丑数 II（LeetCode）

好好学习Py

于 2024-08-02 22:33:41 发布

阅读量364

点赞数 9

分类专栏：算法与数据结构文章标签： leetcode 算法 python 数据结构

本文链接：https://blog.csdn.net/weixin_74254879/article/details/140881630

版权

算法与数据结构专栏收录该内容

82 篇文章 0 订阅

订阅专栏

题目

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

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

解题

def nthUglyNumber(n: int) -> int:
    # 初始化 dp 数组
    dp = [0] * n
    dp[0] = 1

    # 初始化三个指针
    p2 = p3 = p5 = 0

    for i in range(1, n):
        # 计算当前三个候选丑数
        next_ugly = min(dp[p2] * 2, dp[p3] * 3, dp[p5] * 5)

        # 更新 dp 数组
        dp[i] = next_ugly

        # 移动指针
        if next_ugly == dp[p2] * 2:
            p2 += 1
        if next_ugly == dp[p3] * 3:
            p3 += 1
        if next_ugly == dp[p5] * 5:
            p5 += 1

    return dp[n - 1]


# 测试代码
n = 10
print(f"The {n}th ugly number is: {nthUglyNumber(n)}")