STL中multiset和priority_queue的区别以及使用方式

最新推荐文章于 2023-05-17 08:57:32 发布

xmuBin70

最新推荐文章于 2023-05-17 08:57:32 发布

阅读量1.4k

点赞数 2

CC 4.0 BY-SA版权

分类专栏： c++ 文章标签： stl

本文链接：https://blog.youkuaiyun.com/bingoo_t/article/details/109215288

c++ 专栏收录该内容

6 篇文章

订阅专栏

本文探讨了STL中的multiset和priority_queue两种数据结构在建堆过程中的区别，包括处理重复元素的情况。对于堆中有重复元素，multiset能保持元素有序且允许重复，而priority_queue默认不考虑重复。若要创建无重复元素的堆，可以结合set。此外，还介绍了如何使用multiset对链表建堆，以及其在数组去重排序中的作用。

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

multiset和priority_queue 都是内部有序的数据结构，都可以用来实现堆，用法稍有不同。

建堆(大顶堆为例)

堆中有重复元素

使用priority_queue:

priority_queue<int> pq; // 默认大顶堆, 小顶堆: priority_queue<int, vector<int>, greater<int>> pq;
pq.push(1); pq.push(2), ... // 添加元素
while(k-- && !pq.empty()) // 取出第k大的元素，堆内元素会减少
    res = pq.top(), pq.pop();

使用multiset:

multiset<int, greater<int>> ms; // 默认是小根堆，所以需要加greater<int>
ms.insert(1), ms.insert(2), ... // 添加元素
// 取出第k大的数，堆内元素不会减少
for(auto i=ms.begin(); k-- && i!=ms.end(); ++i)
    res = *i;

堆中无重复元素

若希望堆中无重复元素，需要使用set:

set<int, greater<int>> s; // 默认大根堆
s.insert(1), s.insert(2), ...
for(auto i=ms.begin(); k-- && i!=ms.end(); ++i)
    res = *i;

对链表建堆

struct cmp{ // 需要自己编写比较函数
    bool operator()(ListNode* l1, ListNode* l2){
        return l1->val > l2->val;
    }
};
priority_queue<ListNode*, vector<ListNode*>, cmp> lpq;
lpq.push(l1), lp1.push(l2), ...

如果使用multimap则可以免去比较函数的编写

multimap<int, ListNode*, greater<int>> mm; // set和map的默认都是小根堆
mm.insert({l1->val, l1}); // 添加键值对, 按键对值排序
mm.insert({l2->val, l2}); ...

可用于对数组去重排序

// 用set去重, 用assign赋值
set<int> s(v.begin(), v.end());
v.assign(s.begin(), s.end());
// 或者直接用s来初始化v
vector<int> v(s.begin(), s.end());