313. 超级丑数

313. 超级丑数

原始题目链接：https://leetcode.cn/problems/super-ugly-number/

超级丑数 是一个正整数，并满足其所有质因数都出现在质数数组 primes 中。

给你一个整数 n 和一个整数数组 primes ，返回第 n 个 超级丑数 。

题目数据保证第 n 个 超级丑数 在 32-bit 带符号整数范围内。

示例 1：

输入：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] 。
示例 2：

输入：n = 1, primes = [2,3,5]
输出：1
解释：1 不含质因数，因此它的所有质因数都在质数数组 primes = [2,3,5] 中。

提示：

1 <= n <= 105
1 <= primes.length <= 100
2 <= primes[i] <= 1000
题目数据 保证 primes[i] 是一个质数
primes 中的所有值都 互不相同 ，且按 递增顺序 排列

解题思路：

题目的意思是，后面的超级丑数是给定的primes数组中的数乘以前面的超级丑数得到的，使用动态规划解题。

代码实现：

class Solution:
    def nthSuperUglyNumber(self, n: int, primes: List[int]) -> int:
        # 定义数组dp,dp[i]表示第i个丑数
        dp = [0] * (n + 1)
        m = len(primes)
        # pointers表示质因子应该和哪个丑数相乘
        pointers = [0] * m
        # nums表示前面的丑数和primes的元素的结果，1是最小的超级丑数
        nums = [1] * m

        for i in range(1, n + 1):
            # 每次都计质因数和丑数相乘后的算最小值
            min_num = min(nums)
            dp[i] = min_num
            for j in range(m):
                if nums[j] == min_num:
                    pointers[j] += 1
                    nums[j] = dp[pointers[j]] * primes[j]
        
        return dp[n]

参考文献：
https://leetcode.cn/problems/super-ugly-number/solution/chao-ji-chou-shu-by-leetcode-solution-uzff/