[路飞]_leetcode刷题_313. 超级丑数

123 阅读1分钟

题目

超级丑数 是一个正整数,并满足其所有质因数都出现在质数数组 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] 中。

思路:

有了264. 丑数 II的铺垫,超级丑数其实就是把这个解法写成一个通用方案

直接改写一下就出来了。

具体原理,可以去看264. 丑数 II

/**
 * @param {number} n
 * @param {number[]} primes
 * @return {number}
 */
var nthSuperUglyNumber = function(n, primes) {
    let point = new Array(primes.length);
    point.fill(0);
    let dp = [1];
    for(let i=0;i<n;i++){
        let minArr = [];
        
        for(let j=0;j<primes.length;j++){
            minArr.push(dp[point[j]]*primes[j])
        }
        let min = Math.min(...minArr);
        for(let z=0;z<primes.length;z++){
            if(dp[point[z]]*primes[z] == min){
                point[z]++;
            }
        }
        dp.push(min);
    }
    return dp[n-1]
};