刷题使我快乐,满脸开心.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
}