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;如果使用multimap则可以免去比较函数的编写
multimap<int, ListNode*, greater<int>> mm; // set和map的默认都是小根堆 mm.insert({l1->val, l1}); // 添加键值对, 按键对值排序 mm.insert({l2->val, l2}); ...