题意:丑数是指不能被2,3,5以外的其他素数整除的数,求第1500个丑数
解题思路:从小到大生成各个丑数,对于任意丑数x,2x,3x,5x也是丑数,这样就可以用一个优先队列保存suo'y所有已生成的丑数,每次取出最小的丑数生成3个新的丑数。唯一需要注意的是当前的生成的丑数有没有生成过。
AC代码:
#include<stdio.h>
#include<iostream>
#include<vector>
#include<queue>
#include<set>
using namespace std;
typedef long long ll;
const int coeff[3]={2,3,5};
int main()
{
priority_queue<ll,vector<ll>,greater<ll> > pq;
set<ll> s;
pq.push(1);
s.insert(1);
for(int i=1;;i++)
{
ll x=pq.top();
pq.pop();
if(i==1500)
{
cout<<"The 1500'th ugly number is "<<x<<".\n";
break;
}
for(int j=0;j<3;j++)
{
ll tx=x*coeff[j];
if(s.count(tx)==0)
{
pq.push(tx);
s.insert(tx);
}
}
}
return 0;
}