LeetCode 23 合并k个排序链表优先队列堆优化

最新推荐文章于 2022-07-09 15:33:33 发布

TIMELIMITE

最新推荐文章于 2022-07-09 15:33:33 发布

阅读量212

点赞数

分类专栏： LeetCode 链表 Java 文章标签：链表 leetcode 数据结构

本文链接：https://blog.csdn.net/TIMELIMITE/article/details/125576809

版权

LeetCode 同时被 3 个专栏收录

66 篇文章 0 订阅

订阅专栏

Java

54 篇文章 0 订阅

订阅专栏

链表

6 篇文章 0 订阅

订阅专栏

// 合并k个有序链表， 使用优先队列进行优化，复杂度kn * logk
// 熟练掌握java自定义比较器， 也可用lambda表达式

class Solution {

    public ListNode mergeKLists(ListNode[] lists) {

        if (lists == null || lists.length == 0) return null;

        ListNode dummy = new ListNode();

        PriorityQueue<ListNode> que = new PriorityQueue<>(lists.length, new Comparator<ListNode>(){
            public int compare(ListNode a, ListNode b) {
                return a.val - b.val;
            }
        });

//        PriorityQueue<ListNode> que = new PriorityQueue<>(lists.length, (a, b) -> (a.val - b.val));

        for (ListNode list : lists) {
            if (list != null)
                que.add(list);
        }
        ListNode p = dummy;
        while(!que.isEmpty()) {
            ListNode now = que.poll();
            p.next = now;
            if (now.next != null)
                que.add(now.next);
            p = p.next;
        }
        return dummy.next;

    }
}