hiahia_hia1 2023-03-02 15:27 采纳率: 30%
浏览 23
已结题

java算法问题请求解答

#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;
    }
}
  • 写回答

3条回答 默认 最新

  • Huazie 全栈领域优质创作者 2023-03-02 15:35
    关注
    • 使用了 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);
                }
            });
    
    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论 编辑记录
查看更多回答(2条)

报告相同问题?

问题事件

  • 系统已结题 3月10日
  • 已采纳回答 3月2日
  • 创建了问题 3月2日

悬赏问题

  • ¥15 关于#python#的问题
  • ¥15 主机可以ping通路由器但是连不上网怎么办
  • ¥15 数据库一张以时间排好序的表中,找出多次相邻的那些行
  • ¥50 关于DynamoRIO处理多线程程序时候的问题
  • ¥15 kubeadm部署k8s出错
  • ¥15 Abaqus打不开cae文件怎么办?
  • ¥20 双系统开机引导中windows系统消失问题?
  • ¥15 小程序准备上线,软件开发公司需要提供哪些资料给甲方
  • ¥15 关于生产日期批次退货退款,库存回退的问题
  • ¥15 手机应用的时间可以修改吗