java算法问题请求解答

#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;
    }
}
  • 使用了 PriorityQueue 类的 包含比较器的 构造方法:

img

  • 比较器使用了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表达式表示了一个自定义的排序规则,让堆中元素按照数对和进行逆序排序。

不知道你这个问题是否已经解决, 如果还没有解决的话:

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^