从小到大生成各个丑数,最小的丑数是1,而对于任意丑数x,2x、3x和5x也都是丑数。这样,就可以用一个优先队列保存所有已生成的丑数,每次取出最小的丑数,生成3个新的丑数。唯一需要注意的是,同一个丑数有多
种生成方式,所以需要判断一个丑数是否已经生成过。
#include<iostream>
#include<vector>
#include<set>
#include<queue>
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 x2 = x*coeff[j];
if(!s.count(x2)){ pq.push(x2); s.insert(x2); }
}
}
return 0;
}