zcmu1178: идеальное число

Ссылка на заголовок: https://acm.zcmu.edu.cn/JudgeOnline/problem.php?id=1178

Тема

Идеальное число состоит в том, что его простой множитель равен 2, 3, 5, 7 или нет. Что такое n-е совершенное число?

Идеи

Наименьшее совершенное число равно 1, а затем умножьте каждое совершенное число на 2, 3, 5 и 7, тогда результат также будет идеальным числом. Затем сохраните эти числа в очереди приоритетов, а затем отобразите текущее наименьшее число для умножения 2, 3, 5, 7. Вытянутые числа могут быть сохранены в массиве ответов (убедитесь, что данные ответов находятся в порядке возрастания, а не пропущено), и новое поколение Номер добавляется в очередь приоритета.

код переменного тока

#include<bits/stdc++.h>
using namespace std;
#define ll long long
set<ll> s; //判断之前是否出现过
vector<ll> v; //计入答案数组
priority_queue<ll, vector<ll>, greater<ll> >q; //最小堆，堆顶是最小元素
int a[] = {2, 3, 5, 7};
void init(){
	q.push(1);
	for(int i = 1; i <= 5842; i ++){ //一共询问5842
		ll tmp = q.top(); q.pop();
		v.push_back(tmp); //计入答案数组，下标从0开始
		for(int j = 0; j < 4; j ++){
			ll tt = tmp * a[j]; //分别乘2 3 5 7
			if(s.count(tt)) continue; //之前存在过，跳过
			s.insert(tt);
			q.push(tt);
		}
	}
}
int main(){
	init();
	int n; 
	while(~scanf("%d", &n)){
		printf("%d\n", v[n - 1]); //下标从0开始，所以要-1
	}
	return 0;
}