一、题目
给定两个以升序排列的整形数组 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二、思路
建立一个含有两个数值的类,通过两次循环将所有数值的组合构造出来的对象放进一个优先级队列中(PriorityQueue对象默认建立的是最小堆)。通过从优先级队列中循环取出k个最小元素来实现题目要求。
三、代码
class Pair implements Comparable<Pair> {
public int num1;
public int num2;
public int sum;
public Pair(int num1, int num2) {
this.num1 = num1;
this.num2 = num2;
this.sum = num1 + num2;
}
@Override
public int compareTo(Pair o) {
// 用 sum 的值来衡量 Pair 的大小
return this.sum - o.sum;
}
}
public class Solution {
public List<List<Integer>> kSmallestPairs(int[] nums1, int[] nums2, int k) {
List<List<Integer>> result = new ArrayList<>();
if (nums1.length == 0 || nums2.length == 0) {
return result;
}
// 当前采取的是前 k 小的元素,就建立一个小堆
PriorityQueue<Pair> queue = new PriorityQueue<>();
for (int i = 0; i < nums1.length; i++) {
for (int j = 0; j < nums2.length; j++) {
queue.offer(new Pair(nums1[i], nums2[j]));
}
}
for (int i = 0; i < k && !queue.isEmpty(); i++) {
Pair cur = queue.poll();
List<Integer> tmp = new ArrayList<>();
tmp.add(cur.num1);
tmp.add(cur.num2);
result.add(tmp);
}
return result;
}
}