C ++中的超级丑数

我们必须创建一个函数来找到第n个超级丑陋的数字。超级丑数是正数，其所有素数均在给定的素数列表k素数中。因此，如果n为12，质数为[2、7、13、19]，则输出将为32，这是因为[1、2、4、7、8、13、14、16、19、26， [28，32]是12个超级丑陋数字的序列。

为了解决这个问题，我们将遵循以下步骤-

• 创建一个具有num，prim和idx的数据结构三元组

• 如果n为1，则返回1，创建大小为n + 1的数组，并将其填充为1

• 定义优先级队列pq

• 对于i在0到素数大小范围内-

• 创建三元组t（primes [i]，primes [i]，2）

• 对于2到n范围内的i

• curr：= pq的顶部并从pq中删除

• 发生次数：=发生次数* v [发生次数索引]

• 货币指数提高1

• 将curr插入pq

• curr：= pq的顶部元素，然后从pq中删除

• val：=当前数

• v [i]：= val

• 发生次数：=发生次数* v [发生次数索引]

• 货币指数提高1

• 将curr插入pq

• 而val = pq top的数量，

• 返回v [n]

范例（C ++）

让我们看下面的实现以更好地理解-

#include <bits/stdc++.h>
using namespace std;
struct Data{
   int num, prime, idx;
   Data(int a, int b, int c){
      num = a;
      prime = b;
      idx = c;
   }
};
struct Comparator{
   bool operator()(Data a, Data b){
      return !(a.num < b.num);
   }
};
class Solution {
   public:
   int nthSuperUglyNumber(int n, vector<int>& primes) {
      if(n == 1)return 1;
      vector <int> v(n + 1, 1);
      priority_queue < Data, vector < Data >, Comparator > pq;
      for(int i = 0; i < primes.size(); i++){
         pq.push(Data(primes[i], primes[i], 2));
      }
      int x;
      for(int i = 2; i <= n; i++){
         Data curr = pq.top();
         pq.pop();
         int val = curr.num;
         v[i] = val;
         curr.num = curr.prime * v[curr.idx];
         curr.idx++;
         pq.push(curr);
         while(val == pq.top().num){
            curr = pq.top();
            pq.pop();
            curr.num = curr.prime * v[curr.idx];
            curr.idx++;
            pq.push(curr);
         }
      }
      return v[n];
   }
};
main(){
   Solution ob;
   vector<int> v = {2,7,13,19};
   cout << (ob.nthSuperUglyNumber(12, v));
}

输入项

12
[2,7,13,19]

输出结果

32