priority_queue 的用法

本篇博文的重点放在元素比较方式上。
1、比较方式采用默认方式

#include<iostream>
#include<queue>
#include<vector>
#include<ctime>
using namespace std;

priority_queue<int>q1;//默认大的优先
priority_queue<int, vector<int>, greater<int>>q2;//默认小的优先
priority_queue<int, vector<int>, less<int>>q3;   //默认大的优先

void init()
{
    for (int i = 0; i<20; i++)
    {
        //int x=rand()%100;
        q1.push(rand() % 100);
        q3.push(rand() % 100);
        q2.push(rand() % 100);

    }
}
void P(priority_queue<int>q)
{
    while (!q.empty())
    {
        cout << q.top() << " ";
        q.pop();
    }
    cout << endl;
}
void P(priority_queue<int, vector<int>, greater<int> >q)
{
    while (!q.empty())
    {
        cout << q.top() << " ";
        q.pop();
    }
    cout << endl;
}

int main()
{

    srand(time(NULL));
    init();
    P(q1);//(默认)大的优先
    P(q2);//小的优先
    P(q3);//大的优先

    return 0;
}

2、自定义优先级
自定义优先级时,比较函数可以在结构体中定义,此时必须要重载”<”的友元函数,此处有两个重点。
(1)、重载的必须是”<”符合,重载“>”符合会出错。
(2)、必须是友元函数。

#include <iostream>
#include <queue>
#include <ctime>
#include <algorithm>
using namespace std;

struct Node
{
    int x, y;
    Node() {}
    Node( int x,int y):x(x), y(y) {}
    //x小的优先,相同时y小的优先
    friend bool operator<(Node a, Node b)
    {
        if( a.x== b.x ) return a.y> b.y;
        return a.x> b.x;
    }
};
struct cmp
{
    //x小的优先,相同时y小的优先
    bool operator() ( Node a, Node b )
    {
        if( a.x== b.x ) return a.y> b.y;

        return a.x> b.x;
    }
};

int main()
{
    priority_queue<Node, vector<Node>, cmp> q;

    srand(time(NULL));
    for( int i= 0; i< 10; ++i )
        q.push( Node( rand()%10, rand()%10 ) );

    while( !q.empty() )
    {
        cout << q.top().x << ' ' << q.top().y << endl;
        q.pop();
    }
    cout<<"------------------------"<<endl;
    priority_queue<Node>qq;
    for( int i= 0; i< 10; ++i )
        qq.push( Node( rand()%10, rand()%10 ) );

    while( !qq.empty() )
    {
        cout << qq.top().x << ' ' << qq.top().y << endl;
        qq.pop();
    }

    return 0;
}

猜你喜欢

转载自blog.csdn.net/ssf_cxdm/article/details/81557697