力扣 313. 超级丑数

题目

超级丑数 是一个正整数，并满足其所有质因数都出现在质数数组 primes 中。

给你一个整数 n 和一个整数数组 primes ，返回第 n 个 超级丑数 。

题目数据保证第 n 个 超级丑数 在 32-bit 带符号整数范围内。

示例

输入：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] 。

输入：n = 1, primes = [2,3,5]
输出：1
解释：1 不含质因数，因此它的所有质因数都在质数数组 primes = [2,3,5] 中。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/super-ugly-number
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法1：多路归并

如果使用之间的做法 set + 优先队列(解法一) 会超时。

Java实现

class Solution {
    public int nthSuperUglyNumber(int n, int[] primes) {
        int m = primes.length;
        PriorityQueue<int[]> q = new PriorityQueue<>((a, b) -> a[0] - b[0]);

        for (int i = 0; i < m; i++) {
            q.offer(new int[]{primes[i], i, 0});
        }
        
        int[] ans = new int[n];
        ans[0] = 1;
        for (int j = 1; j < n;) {
            int[] cur = q.poll();
            int val = cur[0], i =  cur[1], idx = cur[2];

			//去重
            if (val != ans[j - 1]) ans[j++] = val;
            q.offer(new int[]{primes[i] * ans[idx + 1], i, idx + 1});
        }
        return ans[n - 1];
    }
}