标签：Pairs java get int Sums queue new nums1 nums2
目录
一，题目描述
英文描述
You are given two integer arrays nums1 and nums2 sorted in ascending order and an integer k.
Define a pair (u, v) which consists of one element from the first array and one element from the second array.
Return the k pairs (u1, v1), (u2, v2), ..., (uk, vk) with the smallest sums.
中文描述
给定两个以升序排列的整数数组 nums1 和 nums2 , 以及一个整数 k 。
定义一对值 (u,v)，其中第一个元素来自 nums1，第二个元素来自 nums2 。
请找到和最小的 k 个数对 (u1,v1), (u2,v2) ... (uk,vk) 。
示例与说明
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-k-pairs-with-smallest-sums
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
二，解题思路
java版本的代码采用小根堆，暴力双重循环，将所有组合依次放入堆中，最后筛选前k个；
- 上面的做法会超时，可以对部分条件进行筛选。由于只需要前k个最小组合，每个数组（nums1或nums2）中选出的数字个数不可能超过k（有k个小的不选，选大的？）。这样可以通过目前的测试用例。
C++版本采用大根堆，维持最小的k个组合，堆顶为最小k个组合中最大的一个。
- 当堆的大小达到k时，判断新组合与堆顶的优先级关系，若堆顶优先级高（说明堆顶的组合大于新组合），则将新组合放入堆中，并弹出堆顶。
三，AC代码
C++
class Solution {
public:
struct cmp {
// 大根堆，维持k个最小值，堆顶为最小值中最大的一个
bool operator()(pair<int, int> a, pair<int, int> b) {
return (a.first + a.second) < (b.first + b.second);
}
};
vector<vector<int>> kSmallestPairs(vector<int>& nums1, vector<int>& nums2, int k) {
vector<vector<int>> ans;
priority_queue<pair<int, int>, vector<pair<int, int>>, cmp> queue;// ！！！
for (int i = 0; i < nums1.size() && i < k; i++) {
for (int j = 0; j < nums2.size() && j < k; j++) {
if (queue.size() < k) {
queue.push({nums1[i], nums2[j]});
}
// 当堆顶的优先级大于新的数据时，将其加入到堆中
// 在这里直观的理解就是新数据小于堆顶，因此允许进入前k小
else if (cmp()({nums1[i], nums2[j]}, queue.top())) {
queue.push({nums1[i], nums2[j]});
queue.pop();
}
}
}
while (queue.size()) {
ans.push_back({queue.top().first, queue.top().second});
queue.pop();
}
return ans;
}
};Java
class Solution {
public List<List<Integer>> kSmallestPairs(int[] nums1, int[] nums2, int k) {
List<List<Integer>> ans = new ArrayList<>();
PriorityQueue<List<Integer>> queue = new PriorityQueue<>(new Comparator<>(){
public int compare(List<Integer> a, List<Integer> b) {
return (b.get(0) + b.get(1)) - (a.get(0) + a.get(1));
}
});
// 也可以这样写
// PriorityQueue<List<Integer>> queue = new PriorityQueue<>((a, b)->(
// (b.get(0) + b.get(1) - a.get(0) - a.get(1))
// ));
for (int i = 0; i < nums1.length && i < k; i++) {
for (int j = 0; j < nums2.length && j < k; j++) {
queue.offer(new ArrayList(Arrays.asList(nums1[i], nums2[j])));
if (queue.size() > k) queue.poll();
}
}
while (queue.size() > 0) {
ans.add(queue.poll());
}
return ans;
}
}四，解题过程
第一博
借助优先队列——小根堆，双重循环将每一个可能的结果放进去，然后返回队列中的内容。
class Solution {
public List<List<Integer>> kSmallestPairs(int[] nums1, int[] nums2, int k) {
List<List<Integer>> ans = new ArrayList<>();
PriorityQueue<List<Integer>> queue = new PriorityQueue<>(new Comparator<>(){
public int compare(List<Integer> a, List<Integer> b) {
return (b.get(0) + b.get(1)) - (a.get(0) + a.get(1));
}
});
for (int i = 0; i < nums1.length; i++) {
for (int j = 0; j < nums2.length; j++) {
queue.offer(new ArrayList(Arrays.asList(nums1[i], nums2[j])));
if (queue.size() > k) queue.poll();
}
}
while (queue.size() > 0) {
ans.add(queue.peek());
queue.poll();
}
return ans;
}
}
第二搏
上述解法和暴力解法基本差不多。现在考虑剪枝。
由于nums1和nums2是从小到大的排序数组，所以当双重循环中的指针（i 或 j ）超过k时，就不需要继续计算下去了
第三博
换C++试一试大根堆的方法，直观来说就是维持一个大小为k的堆，堆中存放值最小的元素，这样堆顶就是最小k个元素中，最大的那一个。
当堆的大小达到k，可以将后面的元素与堆定比较，若小于堆顶（也就是堆顶优先级高，这里有点绕），则加入堆中成为前k小的一员，并弹出堆顶。
标签：Pairs,java,get,int,Sums,queue,new,nums1,nums2 来源： https://blog.csdn.net/qq_41528502/article/details/121185394
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。