#leetcode上做了一题,这个是官方题解我没看懂啊,主要是(pair1, pair2) -> pair2.get(0) + pair2.get(1) - pair1.get(0) - pair1.get(1)看不懂,希望有大神能指点一下。
#class Solution {
public ListInteger>> kSmallestPairs(int[] nums1, int[] nums2, int k){
// 创建一个大根堆,堆中元素排序按照数对和进行逆序排序。
PriorityQueueInteger>> heap = new PriorityQueue<>(
(pair1, pair2) -> pair2.get(0) + pair2.get(1) - pair1.get(0) - pair1.get(1)
);
// 数组1取前k个数 (长度n小于k则取n个,数组2也相同),组成 k*k 个数对
int len1 = Math.min(nums1.length, k);
int len2 = Math.min(nums2.length, k);
for(int i = 0; i < len1; i++){
for(int j = 0; j < len2; j++){
// 将数对加入大根堆中
ArrayList<Integer> list = new ArrayList<>();
list.add(nums1[i]);
list.add(nums2[j]);
heap.add(list);
// 如果大根堆中的元素总量超过 k , 则将大根堆的堆顶元素删除。
if(heap.size() > k) heap.poll();
}
}
// 最后剩下的大根堆中的数对就是和最小的k个数对,将其保存到列表中。
ArrayListInteger>> ans = new ArrayList<>();
IteratorInteger>> iterator = heap.iterator();
while (iterator.hasNext()){
ans.add(iterator.next());
}
return ans;
}
}
比较器使用了JDK8的新的特性【Java Lambda 表达式】
上面那个Lambda表达式 创建的 比较器 相当于 如下:
PriorityQueue<List<Integer>> heap = new PriorityQueue<>(new Comparator<List<Integer>>() {
@Override
public int compare(List<Integer> pair1, List<Integer> pair2) {
return pair2.get(0) + pair2.get(1) - pair1.get(0) - pair1.get(1);
}
});
大白话解释:
(pair1, pair2) -> pair2.get(0) + pair2.get(1) - pair1.get(0) - pair1.get(1) 是一个Lambda表达式,表示一个函数。这个函数有两个参数 pair1 和 pair2,它们都是 List 类型的。这个Lambda表达式返回的值是 pair2 和 pair1 数对和的差值,这个差值越小,说明这个数对的和越小,就越应该在堆的前面。
这个Lambda表达式实现了大根堆的排序方式,让大根堆中的元素按照数对和逆序排序。在Java的PriorityQueue中,如果没有指定排序规则,则默认是使用元素的自然顺序排序。但是,如果需要使用一个不同的排序规则,就可以传入一个比较器(Comparator)对象或Lambda表达式来自定义排序规则。在这个例子中,Lambda表达式表示了一个自定义的排序规则,让堆中元素按照数对和进行逆序排序。
不知道你这个问题是否已经解决, 如果还没有解决的话: