leetcode373查找和最小的k对数字

将可能的组合放入priority_queue中，并取最小的k个元素

class Solution {
public:
    vector<pair<int, int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {
        using pr=pair<int,int>;
        vector<pr> res;
        if(nums1.size()<1 ||nums2.size()<1)
            return res;        
        auto cmp=[](pr a,pr b){return a.first+a.second<b.first+b.second;};  //大顶堆
        priority_queue<pr,vector<pr>,decltype(cmp)> pq(cmp);
        for(int i=0;i<min((int)nums1.size(),k);++i)  //输入的范围最多为两个数组各k个元素，共k*k个pair
            for(int j=0;j<min((int)nums2.size(),k);++j)
            {
                pq.push(make_pair(nums1[i],nums2[j]));   
                if(pq.size()>k)   //只存最小的k个
                    pq.pop();
            }
        
        while(!pq.empty())
        {
            res.push_back(pq.top());   //结果中前面的元素大于后面的元素，可以考虑倒序输出
            pq.pop();
        }
        return res;        
    }
};

参考：
https://leetcode.com/problems/find-k-pairs-with-smallest-sums/discuss/84568/c%2B%2B-priority_queue-solution