373. 查找和最小的K对数字 - 力扣(LeetCode)
优先级队列求topK的典型题目,
先来一个暴力的排序法:
class Solution {
public:
vector<vector<int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {
vector<vector<int>> sums;
for(int i=0;i<nums1.size();++i){
for(int j=0;j<nums2.size();++j){
sums.push_back({nums1[i],nums2[j]});//将所有的可能都加进去
}
}
sort(sums.begin(), sums.end(), [](vector<int>& l, vector<int>& r){return (l[0]+l[1] < r[0]+r[1]);});
if(k > sums.size()) return sums;
vector<vector<int>> res(sums.begin(), sums.begin()+k);
return res;
}
};