STL中multiset和priority

it2025-08-10  9

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());
最新回复(0)