合并K个有序链表Merge k Sorted Lists

最新推荐文章于 2024-07-30 23:25:53 发布

Eric.Cui

最新推荐文章于 2024-07-30 23:25:53 发布

阅读量66

点赞数

分类专栏：数据结构与算法文章标签：链表

本文链接：https://blog.csdn.net/edisonzhi/article/details/117924337

版权

数据结构与算法专栏收录该内容

201 篇文章 0 订阅

订阅专栏

本文介绍了如何通过优先队列实现合并K个已排序的链表，创建一个单一的升序链表。核心思路是在队列中保持最小元素在前，不断取出最小节点并连接成新的有序链表，同时将后续节点加入队列，直至队列为空，最终返回合并后的链表头节点。

摘要由CSDN通过智能技术生成

合并K个有序链表Merge k Sorted Lists

有一个链表数组，每个链表都是有序的【升序排列】。合并所有链表到一个升序链表中。

思路

N路有序链表合并，借助优先队列，维持一个容量为N的队列，每次出队最小的，然后再入队一个元素。

public ListNode mergeKLists(ListNode[] lists) {
        if(lists.length==0){
            return null;
        }
        ListNode h = new ListNode(0);
        ListNode p = h;

        PriorityQueue<ListNode> pq = new PriorityQueue<>((o1, o2) -> o1.val - o2.val);
        for (ListNode x : lists) {
           if (x!=null){
               pq.offer(x);
           }
        }

        while (!pq.isEmpty()){
            ListNode cur = pq.peek();
            p.next = cur;
            pq.poll();
            p = cur;
            if (cur.next!=null){
                pq.offer(cur.next);
            }
        }
        return h.next;
    }