LeetCode 第23题:合并K个升序链表
题目描述
给你一个链表数组,每个链表都已经按升序排列。请你将所有链表合并到一个升序链表中,返回合并后的链表。
难度
困难
题目链接
https://leetcode.cn/problems/merge-k-sorted-lists/
示例
示例 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 = []
输出:[]
示例 3:
输入:lists = [[]]
输出:[]
提示
- k == lists.length
- 0 <= k <= 10^4
- 0 <= lists[i].length <= 500
- -10^4 <= lists[i][j] <= 10^4
- lists[i] 按升序排列
- lists[i].length 的总和不超过 10^4
解题思路
方法一:分治合并
将K个链表的合并问题分解为两两合并的子问题。
关键点:
- 使用分治思想,将K个链表分成两部分
- 递归地合并每部分的链表
- 最后将两个合并后的链表合并
- 复用第21题的两个链表合并方法
具体步骤:
- 如果链表数组为空,返回null
- 如果只有一个链表,直接返回
- 将链表数组分成两半
- 递归合并左半部分和右半部分
- 合并得到的两个链表
时间复杂度:O(N * log k),其中k是链表数量,N是所有节点的总数
空间复杂度:O(log k),递归栈的深度
方法二:优先队列
使用优先队列(最小堆)来维护k个链表的当前最小节点。
思路:
- 创建最小堆,将k个链表的头节点加入堆中
- 每次从堆中取出最小节点,加入结果链表
- 如果被取出的节点有后继节点,将后继节点加入堆中
- 重复步骤2-3直到堆为空
代码实现
C# 实现(分治合并)
/**
* Definition for singly-linked list.
* public class ListNode {
* public int val;
* public ListNode next;
* public ListNode(int val=0, ListNode next=null) {
* this.val = val;
* this.next = next;
* }
* }
*/
public class Solution {
public ListNode MergeKLists(ListNode[] lists) {
if (lists == null || lists.Length == 0) return null;
return