合并K个升序链表——优先级队列

原创已于 2022-02-09 15:52:42 修改 · 267 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#链表 #数据结构 #算法 #队列

于 2022-02-09 15:44:32 首次发布

leetcode 专栏收录该内容

7 篇文章

订阅专栏

该博客介绍了一种利用优先级队列（PriorityQueue）合并多个已排序链表的方法，以创建一个新的有序链表。通过将链表的头节点入队，并按节点值比较优先级，不断取出最小值的节点拼接成新的链表，直至队列为空。示例展示了如何处理包含空链表的情况，并给出了完整的Java代码实现。

描述

给你一个链表数组，每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中，返回合并后的链表。

示例 1：

输入：lists = [[1,4,5],[1,3,4],[2,6]]
输出：[1,1,2,3,4,4,5,6]
解释：链表数组如下：
[1->4->5,1->3->4, 2->6]
将它们合并到一个有序链表中得到。
1->1->2->3->4->4->5->6

示例 2：

输入：lists = []
输出：[]

说明

队列是一种特征为FIFO的数据结构，每次从队列中取出的是最早加入队列中的元素。但是，许多应用需要另一种队列，每次从队列中取出的应是具有最高优先权的元素，这种队列就是优先级队列(Priority Queue)，也称为优先权队列。

代码

 /**
     * 合并K个有序链表
     *
     * @param lists 链表数组
     * @return 合并后的链表
     */
    public ListNode mergeKLists(ListNode[] lists) {
        if (lists == null || lists.length == 0) return null;//空值直接返回
        //实例化优先级队列
        PriorityQueue<ListNode> queue = new PriorityQueue<>(lists.length, Comparator.comparingInt(o -> o.val));
        ListNode result = new ListNode(0);//用于保存结果
        ListNode temp = result;//临时变量，temp引用对象指向result
        for (ListNode node : lists) {//链表入队列
            if (node != null) queue.add(node);
        }
        while (!queue.isEmpty()) {
            temp.next = queue.poll();//优先级最大的出队列，即队列中头节点最小的链表
            temp = temp.next;//移动链表指针
            if (temp.next != null) queue.add(temp.next);//将移动后的链表重新入队列
        }
        return result.next;
    }

   public class ListNode {
        int val;
        ListNode next;

        ListNode() {
        }

        ListNode(int val) {
            this.val = val;
        }

        ListNode(int val, ListNode next) {
            this.val = val;
            this.next = next;
        }
  }