[路飞]_算法_ 超级丑数——动态规划

116 阅读2分钟

题目描述

超级丑数 是一个正整数,并满足其所有质因数都出现在质数数组 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 <= 106
  • 1 <= primes.length <= 100
  • 2 <= primes[i] <= 1000
  • 题目数据 保证 primes[i] 是一个质数
  • primes 中的所有值都 互不相同 ,且按 递增顺序 排列

来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/su… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路

此题是上一题丑数二,的升级版,也可以用暴力解锁,和动态规划解决
上题已经详细讲解,可看丑数二 但此题primes可能很复杂,暴力解题超出时间限制,所以用动态规划解决。
思路一样,不过质因数不是固定了,指针也不是固定的p2,p3,p5,而是采用动态生成
详见代码

代码

/**
 * @param {number} n
 * @param {number[]} primes
 * @return {number}
 */
var nthSuperUglyNumber = function(n, primes) {
    if(n===1) return 1;
    const dp=new Array(n+1).fill(0);
    dp[1]=1;
    let px=[];//指针列表
    for(i=0;i<primes.length;i++){
        px.push({
            val:primes[i],
            index:1
        })
    };

    for(let i=2;i<=n;i++){
        let nums=[],min;//nums为根据指针计算出的待选丑数,min记录最小的丑数也是这一轮的dp[i]
        for(let j=0;j<px.length;j++){
            const value=dp[px[j].index]*px[j].val;
           nums.push(value);
           min=min?Math.min(min,value):value;
       }; 
        dp[i]=min;
        for(let j=0;j<nums.length;j++){
         // 已生成丑数的指针+1
            if(nums[j]===min){
               px[j].index+=1;
            }
       }; 
    }

    return dp[n]
    
};