[C++] 数据结构 priority_queue

优先队列（priority_queue）其实就是堆，默认是大顶堆，即 top() 会返回堆顶元素（最大）。

// 默认大顶堆
priority_queue<int> que;
// 小顶堆
priority_queue<int, vector<int>, greater<int>> que;

如果存储的是 pair 类型，默认会使用 pair 中的第一个元素进行比较，第一个元素相等你则会比较第二个。

// 默认大顶堆
priority_queue<pair<int,int>> que;
// 小顶堆
priority_queue<pair<int,int>, vector<pair<int,int>>, greater<pair<int,int>>> que;

如果想要自定义比较函数，比较麻烦…

class Solution {
    
    
public:
    vector<vector<int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {
    
    
        auto cmp = [&nums1, &nums2](const pair<int, int>& a, const pair<int, int>& b) {
    
    
            return nums1[a.first] + nums2[a.second] > nums1[b.first] + nums2[b.second];
        };
        priority_queue<pair<int, int>, vector<pair<int, int>>, decltype(cmp)> pq(cmp);
};