题目:Ugly Numbers
题意:丑数是指不能被2,3,5以外的其他素数整除的数。把丑数从小到大排列起来,结果如下:
1,2,3,4,5,6,8,9,10,12,15,…
求第1500个丑数。
思路:丑数乘2,3,5还是丑数,最小的丑数是1,所以用1乘上2,3,5,然后存放在优先队列中,每次以最小的出队进行乘2,3,5再存入,注意在存放的是否用set查下重复的!
参考:入门经典-例5-7-P120
代码:
#include <iostream>
#include <vector>
#include <queue>
#include <set>
#include <cstdio>
using namespace std;
typedef long long LL;
int coeff[3] = {2,3,5};
priority_queue<LL,vector<LL>,greater<LL> >pq;//以小的为优先级高的优先队列
set<LL>s;
int main()
{
pq.push(1);
s.insert(1);
for(int i=1;;i++)
{
LL x = pq.top();
pq.pop();
if(i == 1500)
{
printf("The 1500'th ugly number is %lld.\n",x);
break;
}
for(int j=0;j<3;j++)
{
LL temp = x*coeff[j];
if(!s.count(temp))
{
s.insert(temp);
pq.push(temp);
}
}
}
return 0;
}