LeetCode 313 超级丑数

LeetCode 313 超级丑数

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.题解：

方法:优先队列
本题将丑数2的3个质因子拓展到了 K个质因子。
要计算从小到大 第N个

1. 创建K个指针，并标记指针对应的序列数
2. 创建一个最小堆，每次去堆顶元素，并入堆一个当前序列的下一个元素
3. 考虑去重

时间复杂度： O（NlogK）产生一个数的时间复杂度是logK，所以总和为 NlogK
空间复杂度： O(N +K ), 堆与数组的空间的总和。

C++

class Solution {
public:
    int nthSuperUglyNumber(int n, vector<int>& primes) {
        vector<int> s(n) ;
        s[0] = 1; 
        typedef pair<int, int> PII ;
        priority_queue<PII, vector<PII>, greater<PII>>heap ;
        for (auto x: primes) heap.push({s[0]*x, 0}) ;
        for (int idx = 1; idx <n; ) {
            auto t = heap.top();
            heap.pop() ;
            auto val = t.first, ptr = t.second ;//元素的值，以及元素所属序列的排序
            auto x = val / s[ptr] ; //就算元素所属的序列
            if (val != s[idx-1]) s[idx++] =  val ; //若不等于上一个丑数，idx++
            heap.push({(long long)s[ptr + 1] * x, ptr + 1}); //操作就是 将 这个元素 所属序列 的 这个元素 后面一个 元素 加入 堆中进行排序.
        }
        return s[n-1];
    }
};

原题：超级丑数