丑数

丑数

 

 

STL的优先队列定义在<queue>中,用“priority_queue<int>pq”来声明。这个pq是越小的整数优先级越低的队列,由于出队元素不是先入队的元素,所以front()变为了top()

越小的整数优先级越大"priority_queue<int,vector<int>,greater<int> >pq",注意最后的两个>需要用空格隔开,否则会被认为是">>"运算符

 

丑数:设一个丑数为x,那么2x,3x,5x都为丑数,所以从1开始,逐步推出丑数放在集合和优先队列中,每次推导时,取出队列中最小的值,然后删去,依次类推

#include<cstdio>
#include<iostream>
#include<vector>
#include<queue>
#include<set>
using namespace std;


const int a[3] = { 2,3,5 };
int main(void)
{
    typedef long long LL;
    priority_queue<LL, vector<LL>, greater<LL> > pq;
    set<LL> stock;
    stock.insert(1);
    pq.push(1);
    for (int i = 1;; i++)
    {
        LL t = pq.top();
        pq.pop();
        if (i == 1500)
        {
            cout << "The 1500'th ugly number is " << t << "." << endl ;
            break;
        }

        else
        {
            for (int j = 0; j < 3; j++)
            {
                LL x = t * a[j];
                if (!stock.count(x))
                {
                    stock.insert(x);
                    pq.push(x);
                }
            }
        }
    }
    return 0;
}

 

上一篇:MySQL入门(7)——表数据的增、删、改


下一篇:合并果子+c++优先队列priority_queue的用法