23. 合并 K 个升序链表

目录

一、问题描述

二、解题思路

三、代码

四、复杂度分析

---

一、问题描述

给你一个链表数组，每个链表都已经按升序排列。

请你将所有链表合并到一个升序链表中，返回合并后的链表。

二、解题思路

分治法

1. 两两合并：通过两两合并的方法逐步合并所有链表。每次合并两个链表，最终将所有链表合并为一个。
2. 递归合并：可以采用递归的方式来两两合并，先将链表数组分为两半，递归合并左半部分和右半部分，最后合并两个部分。
3. 时间复杂度：分治法的时间复杂度同样是 O(n log k)，但递归的实现方式较为清晰。

三、代码

/**
 * Definition for singly-linked list.
 * 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; }
 * }
 */
class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        // 使用分治法两两合并
        if (lists == null || lists.length == 0) {
            return null;
        }
        return mergeLists(lists, 0, lists.length - 1);
    }

    // 递归分治合并函数
    private ListNode mergeLists(ListNode[] lists, int left, int right) {
        if (left == right) {
            return lists[left];
        }
        int mid = left + (right - left) / 2;
        ListNode l1 = mergeLists(lists, left, mid);
        ListNode l2 = mergeLists(lists, mid + 1, right);
        return mergeTwoLists(l1, l2);
    }

    // 合并两个有序链表的函数
    private ListNode mergeTwoLists(ListNode l1, ListNode l2) {
        ListNode dummy = new ListNode(0);
        ListNode current = dummy;
        
        while (l1 != null && l2 != null) {
            if (l1.val <= l2.val) {
                current.next = l1;
                l1 = l1.next;
            } else {
                current.next = l2;
                l2 = l2.next;
            }
            current = current.next;
        }

        // 连接剩余的节点
        if (l1 != null) {
            current.next = l1;
        } else if (l2 != null) {
            current.next = l2;
        }
        return dummy.next;
    }
}

四、复杂度分析

• 时间复杂度：O(n log k)，因为需要进行 log k 次合并操作，每次合并操作的时间复杂度为 O(n)。
• 空间复杂度：O(log k)，由于递归调用的深度为 log k。