【路飞】算法与数据结构-超级丑数

125 阅读2分钟

不管全世界所有人怎么说,我都认为自己的感受才是正确的。无论别人怎么看,我绝不打乱自己的节奏。喜欢的事自然可以坚持,不喜欢的怎么也长久不了。

LeetCode:原题地址

题目要求

超级丑数 是一个正整数,并满足其所有质因数都出现在质数数组 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 中的所有值都 互不相同 ,且按 递增顺序 排列

思路

res数组保存所有超级丑数。 用pointers数组记录各个质因子对应的指针(有几个质因子就有几个指针),指向res下标,初始都指向下标1(废弃res[0])。 用nums数组记录 指针指向的res中的超级丑数*对应的质因子 乘积。 取nums中最小的数作为当前res元素。看看这个元素是由哪个质因子乘积来的,相应的pointers中的指针++往前移一个

代码

/**
 * @param {number} n
 * @param {number[]} primes
 * @return {number}
 */
var nthSuperUglyNumber = function(n, primes) {
    let res=new Array(n+1).fill(1),pointers=new Array(primes.length).fill(1),nums=[];
    // 外层循环遍历填充res
    for(let i=2;i<=n;i++){
        // 内层循环1:计算当前 指针指向的丑数*指针对应的质因子
        for(let j=0;j<primes.length;j++){
            nums[j]=res[pointers[j]]*primes[j];
        }
        res[i]=Math.min(...nums);
        // 内层循环2:找到选出的res[i]对应的指针,把指针往前移
        for(let k=0;k<nums.length;k++){
            if(nums[k]==res[i]) pointers[k]++;
        }
        nums=[];
    }
    return res[n];
};