LeetCode[Python] 313.超级丑数

最新推荐文章于 2021-08-09 11:07:51 发布

咖喱咖喱饭

最新推荐文章于 2021-08-09 11:07:51 发布

阅读量744

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/j55661132/article/details/82179274

版权

LeetCode 专栏收录该内容

9 篇文章 0 订阅

订阅专栏

LeetCode 313.超级丑数

编写一段程序来查找第 n 个超级丑数。

超级丑数是指其所有质因数都是长度为 k 的质数列表 primes 中的正整数。

示例:

输入: n = 12, primes = [2,7,13,19]
输出: 32
解释: 给定长度为 4 的质数列表 primes = [2,7,13,19]，前 12 个超级丑数序列为：[1,2,4,7,8,13,14,16,19,26,28,32] 。

说明:

1 是任何给定 primes 的超级丑数。
给定 primes 中的数字以升序排列。
0 < k ≤ 100, 0 < n ≤ 106, 0 < primes[i] < 1000 。
第 n 个超级丑数确保在 32 位有符整数范围内。

思路：

首先给定一个解题基本思路：
假设为丑数， $M_{1} = 2 * M, M_{2} = 3*M, M_{3} = 5* M$ 则可以认为 $M_{1}, M_{2}, M_{3}$ 皆为丑数。
切下一丑数最小值一定为 $M_{2}$

依照上述定义可以使用动态规划进行题解。

代码：

class Solution:
    def nthSuperUglyNumber(self, n, primes):
        """
        :type n: int
        :type primes: List[int]
        :rtype: int
        """
        dp = [1]

        lenPrimes = len(primes)
        idxPrimes = [0] * lenPrimes
        counter = 1
        while counter < n:
            min = pow(2, 32)
            for i in range(0, lenPrimes):
                temp = dp[idxPrimes[i]] * primes[i]
                if temp < min:
                    min = temp

            for i in range(0, lenPrimes):
                if min == dp[idxPrimes[i]] * primes[i]:
                    idxPrimes[i] += 1
            dp.append(min)
            counter += 1

        return dp[counter - 1]