LeetCode 23. Merge k Sorted Lists

最新推荐文章于 2025-05-03 16:06:16 发布

原创最新推荐文章于 2025-05-03 16:06:16 发布 · 177 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种使用优先队列高效合并K个已排序链表的方法，通过实例展示输入多个排序链表如何输出单一排序链表。文章详细解释了算法实现过程，包括初始化优先队列、遍历链表并将节点添加到队列中，以及从队列中取出最小节点并连接到结果链表的过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.

Example:

Input:
[
  1->4->5,
  1->3->4,
  2->6
]
Output: 1->1->2->3->4->4->5->6

使用优先队列可以很方便的解决这个问题

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        if(lists.length==0) return null;
        
        PriorityQueue<ListNode> pq = new PriorityQueue<>( (node1,node2)->node1.val-node2.val );//java8 comparator
        ListNode dummy = new ListNode(0);
        ListNode tail = dummy;
        for(ListNode node:lists){
            if(node!=null){
                pq.offer(node);
            }
        }
        
        while(!pq.isEmpty()){
            tail.next = pq.poll();
            tail = tail.next;
            
            if(tail.next!=null){
                pq.offer(tail.next);
            }
        }
        return dummy.next;
        
    }
}