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

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

建堆(大顶堆为例)

堆中有重复元素

使用priority_queue:

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

使用multiset:

multiset<int, greater<int>> ms; // 默认是小根堆，所以需要加greater
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());