算法入门竞赛经典 例题5-7 丑数

丑数：只包含因子 2、 3、 5 的数称为丑数，例如 2 、 3 、 4 、 5 都是都是丑数，但是 7 、 14  不是丑数。现在求从小到大的第1500 个丑数，我们把 1 作为第一个丑数。————  源自《剑指offer》

代码：

#include<iostream>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std;
typedef long long ll;
const int coeff[3]={2,3,5};

int main()
{
    priority_queue<ll,vector<ll>,greater<ll>>pq;
    vector<ll>s;
    pq.push(1);
    s.push_back(1);
    for(int i=1;;i++)
    {
        ll x=pq.top();pq.pop();
        if(i==1500){cout<<x<<endl;break;}
        for(int j=0;j<3;j++)
        {
            ll x2=x*coeff[j];
            if(count(s.begin(),s.end(),x2)==0){s.push_back(x2);pq.push(x2);}
        }
    }
    return 0;
}

TIPS：根据丑数的定义，我们可以知道丑数可以由另外一个丑数乘以2，3或者5得到。因此我们可以创建一个数组，里面的数字是排好序的丑数，每一个丑数都是前面的丑数乘以2，3或者5得到的。