java算法问题请求解答

Question

#leetcode上做了一题，这个是官方题解我没看懂啊，主要是(pair1, pair2) -> pair2.get(0) + pair2.get(1) - pair1.get(0) - pair1.get(1)看不懂，希望有大神能指点一下。

#class Solution {
    public List<List<Integer>> kSmallestPairs(int[] nums1, int[] nums2, int k){
        // 创建一个大根堆，堆中元素排序按照数对和进行逆序排序。
        PriorityQueue<List<Integer>> 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个数对，将其保存到列表中。
        ArrayList<List<Integer>> ans = new ArrayList<>();
        Iterator<List<Integer>> iterator = heap.iterator();
        while (iterator.hasNext()){
            ans.add(iterator.next());
        }
        return ans;
    }
}

Answer 1

• 使用了 PriorityQueue 类的 包含比较器的 构造方法：

• 比较器使用了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);
            }
        });