【算法常用STL】-＞ priority_queue

顾冷__

已于 2022-11-04 18:56:32 修改

阅读量8.2k

点赞数 13

CC 4.0 BY-SA版权

分类专栏： STL 文章标签：算法 leetcode 职场和发展

于 2022-03-11 16:35:54 首次发布

本文链接：https://blog.youkuaiyun.com/GuLeng______/article/details/123422057

STL 专栏收录该内容

11 篇文章

订阅专栏

本文介绍了C++中priority_queue的数据结构，包括如何通过less和greater来创建大根堆和小根堆，以及自定义比较函数的使用方法。通过实例演示了如何在ListNode*结构上构建小顶堆，并提供了相关例题来加深理解。

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

优先队列 Heap

priority_queue<int, vector<int>, less<int>>s;//less表示按照递减(从大到小)的顺序插入元素
priority_queue<int, vector<int>, greater<int>>s;//greater表示按照递增（从小到大）的顺序插入元素

不写第三个参数或者写成less都是大根堆。greater是小根堆。

在C++中优先队列默认的是大根堆，如果用小根堆则加入greater.

定义方式

priority_queue<ListNode * , vector<ListNode *> ,Cmp> heap;

构造小根堆（重载）

1. 自定义结构重载

struct Cmp
    {
        bool operator()(ListNode * a, ListNode * b)
        {
            return a->val > b ->val; //小顶堆，，改为 < 是大顶堆
        }

    };
priority_queue<ListNode * , vector<ListNode *> ,Cmp> heap;

2. 重载 < >

如果是小根堆，就要重载 >
如果是大跟读, 就要重载 <

struct Node{
    int x,y,z;
    friend bool operator > (const Node& a,const Node& b){
        return a.x > b.x; //greater,小顶堆
    } 
    friend bool operator < (const Node& a,const Node& b){
        return a.x < b.x;//less，大顶堆 
    } 
}