题目:
超级丑数 是一个正整数,并满足其所有质因数都出现在质数数组 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] 中。
代码:
class Solution:
def nthSuperUglyNumber(self, n: int, primes: List[int]) -> int:
dp=[0 for _ in range(n+1)]
dp[1]=1
#site用于记录primes中的质数下次乘dp中得哪个位置,如果site[1]=4,则代表dp列表的dp[4]下次和primes的[1]相乘也就是(dp[site[k]]*primes[k])
site=[1 for _ in range(len(primes))]
for x in range (2,len(dp)):
min_num = min(dp[site[k]]*primes[k] for k in range(len(primes)))
dp[x]=min_num
#用于检查是否会出现相同的情况,除重。
for k in range(len(primes)):
if dp[site[k]]*primes[k] == dp[x]:
site[k]+=1
return dp[n]