LeetCode每日一题 313.超级丑数

459 阅读2分钟

这是我参与8月更文挑战的第9天,活动详情查看:8月更文挑战

313. 超级丑数

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

方法一

多指针:此题与丑数的区别就是,丑数是确定了包含给出的固定的质因子,而本题是动态的给出质因子;在丑数中,我们可以根据给出的固定数量的质因子来人为的创建指针;那么,在本题中,质因子数量可变,我们开一个数组来存对应的指针即可;

  • 首先根据primes的大小,开辟一个大小一样的指针数组points
  • res数组记录从小到大的n个数
  • 将1加入res数组中,各个质因子的指针指向1
  • 每次选取质因子和其指针指向的数的乘积的最小值
  • 将该值放入res数组中
  • 最后,将所有乘积与放入数组中一样的值的指针后移一位
class Solution {
    public int nthSuperUglyNumber(int n, int[] primes) {
        int len = primes.length, idx = 0;
        int[] points = new int[len], res = new int[n];
        res[idx ++] = 1;
​
        while(idx < n) {
            int t = Integer.MAX_VALUE;
            for (int i = 0; i < len; i ++ )
                t = Math.min(t, primes[i] * res[points[i]]);
                
            for (int i = 0; i < len; i ++ ) 
                if (t == primes[i] * res[points[i]])
                    points[i] ++;
            res[idx ++] = t;
        }
        for (int i = 0; i < n; i ++ ) System.out.print(res[i] + " ");
        return res[n - 1];
    }
}