刷题使我快乐,满脸开心.jpg
- 来源:力扣(LeetCode)
- 链接: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
中的所有值都互不相同
,且按递增顺序
排列
思路
这道题与丑数ii思路基本相同,而且没有丑数iii的坑拌脚,只是从三个质因数变成了n
个,依法炮制即可
代码
func nthSuperUglyNumber(n int, primes []int) int {
dp := make([]int, n+1)
dp[1] = 1
pointers := make([]int, len(primes))
for k := range pointers {
pointers[k] = 1
}
for i := 2; i <= n; i++ {
minNum := math.MaxInt
for j, prime := range primes {
minNum = min(minNum, dp[pointers[j]]*prime)
}
dp[i] = minNum
for j, prime := range primes {
if minNum == dp[pointers[j]]*prime {
pointers[j]++
}
}
}
return dp[n]
}
func min(a, b int) int {
if a < b {
return a
}
return b
}