priority_queue 的用法

最新推荐文章于 2025-04-18 15:23:29 发布

转载最新推荐文章于 2025-04-18 15:23:29 发布 · 102 阅读

文章标签：

#priority_queue #C++ #大小堆

本文介绍C++中优先队列的使用方法，并演示如何通过自定义比较函数实现不同排序逻辑。首先展示了如何创建默认优先级的大顶堆和小顶堆，并随机填充数据进行展示。随后介绍了自定义结构体及重载小于号运算符来实现特定的优先级规则。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

本篇博文的重点放在元素比较方式上。
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;
}