Leetcode 264. 丑数 II

给你一个整数 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来做,首先创建一个初始的数组,数组起始元素均为1,然后定义三个元素用来2、3、5的计数,遍历dp中的所有元素,每个元素都取对应元素个数乘完2、3、5后的最小值,直到给第十个元素赋值即为第十个丑数。

class Solution:
    def nthUglyNumber(self, n: int) -> int:
        if n < 0:
            return 
        index2 = 0
        index3 = 0
        index5 = 0
        dp = [1 for i in range(n)]
        for i in range(1, n):
            dp[i] = min(dp[index2] * 2, dp[index3] * 3, dp[index5] * 5)
            if dp[i] == dp[index2] * 2:
                index2 += 1
            if dp[i] == dp[index3] * 3:
                index3 += 1
            if dp[i] == dp[index5] * 5:
                index5 += 1
        return dp[n-1]

Leetcode 264. 丑数 II_第1张图片

你可能感兴趣的:(leetcode练习,python,leetcode)