• C++ priority_queue的自定义比较方式


    less对应“<”运算符,

    greater对应">"运算符。

    最近学习STL,发现STL默认都是使用()比较的,默认比较使用less(即'<'运算符),如sort(a,a+n),默认将数组按照递增的顺序来排序(前面的元素<后面的嘛),但是优先队列的源码比较奇特,虽然按道理使用less比较应该默认是小根堆(即堆顶元素最小),但是priority_queue<int, vector<int>, less<int> >却是大根堆,这是因为优先队列队首指向最后,队尾指向最前面的缘故,所以每次使用top()返回的都是队尾的元素。

    对于结构体类型,如果想自定义比较方式,可以写成以下方式,这里要注意的就是priority_queue每次都是从队尾取元素的,所以使用top()返回的是排序后的最后一个元素

    #include <iostream>
    #include <stdio.h>
    #include <string.h> 
    #include <queue>
    #include <functional>
    
    using namespace std;
    
    struct node
    {
        int a,b;
        int cost;
        
    } nod[100];
    
    
    //自定义比较类 
    struct cmp
    {
        //重载()运算符 
        bool operator()(const node& n1, const node& n2)
        {
            //比较操作,按cost从小到大排序,但是这里构造出来的堆每次使用top()返回的都是cost最大的那个元素,因为堆每次都是从队尾弹出元素 
            return n1.cost < n2.cost;
        }
    };
    
    int main()
    {
        priority_queue<node, vector<node>, cmp > q;
        for(int i = 0; i < 5; ++i)
        {
            cin >> nod[i].a >> nod[i].b >> nod[i].cost;
            q.push(nod[i]);
        }
        
        while(!q.empty())
        {
            node x = q.top();
            cout << x.a << ' ' <<x.b << ' ' <<x.cost << endl;
            q.pop();
        }
        
        return 0;
    }

    参考:https://blog.csdn.net/aamahone/article/details/82787184

  • 相关阅读:
    Codeforces Round #535 (Div. 3)
    2019 CCPC-Wannafly Winter Camp Day4(Div2, onsite)
    Codeforces Round #534 (Div. 2)
    2019 CCPC-Wannafly Winter Camp Day3(Div2, onsite)
    2019 CCPC-Wannafly Winter Camp Day2(Div2, onsite)
    2019 CCPC-Wannafly Winter Camp Day1 (Div2, onsite)
    codeforces1097D Makoto and a Blackboard 数学+期望dp
    【NOIP2016】换教室
    ICPC2019徐州站游记
    【Codeforces】Orz Panda Cup
  • 原文地址:https://www.cnblogs.com/FengZeng666/p/12821842.html
Copyright © 2020-2023  润新知