leetcode_313. 超级丑数

题目链接


喜极而泣，刷力扣还是有效果的，今天这个题没有看题解做出来了，再一看和官方题解用最小堆的思路一致，由此联想到做丑数II时确实也是用的这种方法，今天就有种信手拈来的感觉了呜呜（当然值得一提的是看了一眼提示有优先队列才开始联想的qwq）

核心思想如下：（摘自官方题解）

1. 要得到从小到大的第 n个超级丑数，可以借助最小堆实现。
2. 初始时堆为空。首先将最小的超级丑数 1加入堆。每次取出堆顶元素 x，则 x 是堆中最小的超级丑数。对于数组 primes 的任意质数 p，p*x 也是超级丑数，因此将数组primes 中的每个质数和x 的乘积分别加入堆。
3. 上述做法会有重复元素加入堆中，如样例一的[2,7,13,19], 27和72就是会重复的，可以采用set去重，也可以获取堆顶元素x的时候使用while循环判断取出的元素x是否还和堆顶元素相同，相同的话就一直pop，也能实现去重。

代码：

class Solution {
public:
    int nthSuperUglyNumber(int n, vector<int>& primes) {
        //结合set进行去重
        priority_queue<long long, vector<long long>, greater<long long> > q;
        set<long long> s;
        if(n==1) return 1;
        int len=primes.size();
        q.push(1);
        int prime=0;
        while(n--){
            prime=q.top();
            q.pop();
            if(n==0) break;
            for(int i=0;i<len;i++){
                long long tmp=(long long)prime*primes[i];
                if(!s.count(tmp)){
                    s.insert(tmp);
                    q.push(tmp);
                }
            }
        }
        return prime;
    }
};