将可能的组合放入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;
}
};