LeetCode-23. 合并K个排序链表

最新推荐文章于 2024-10-01 18:51:18 发布

遗忘的大城小事

最新推荐文章于 2024-10-01 18:51:18 发布

阅读量108

点赞数

分类专栏： LeetCode 文章标签：链表 java 归并排序

本文链接：https://blog.csdn.net/qq_34449717/article/details/105774938

版权

LeetCode 专栏收录该内容

135 篇文章 0 订阅

订阅专栏

题目

合并 k 个排序链表，返回合并后的排序链表。请分析和描述算法的复杂度。

示例

输入:
[
1->4->5,
1->3->4,
2->6
]
输出: 1->1->2->3->4->4->5->6

解题思路

思想：分治合并。

将 $k$ 个链表配对并将同一对中的链表合并；
第一轮合并以后， $k$ 个链表被合并成了 $\frac{k}{2}$ 个链表，平均长度为 $\frac{2n}{k}$ ，然后是 $\frac{k}{4}$ 个链表， $\frac{k}{8}$ 个链表等；
重复这一过程，直到链表整个有序。

代码实现

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        return merge(lists, 0, lists.length - 1);
    }

    public ListNode merge(ListNode[] lists, int low, int high) {
        if (low == high) {
            return lists[low];
        }
        if (low > high) {
            return null;
        }
        int mid = low + (high - low) / 2;
        ListNode listNode1 = merge(lists, low, mid);
        ListNode listNode2 = merge(lists, mid + 1, high);
        return merge2Lists(listNode1, listNode2);
    }

    public ListNode merge2Lists(ListNode listNode1, ListNode listNode2) {
        ListNode head = new ListNode();
        ListNode rear = head;
        while (listNode1 != null && listNode2 != null) {
            if (listNode1.val < listNode2.val) {
                rear.next = listNode1;
                listNode1 = listNode1.next;
            } else {
                rear.next = listNode2;
                listNode2 = listNode2.next;
            }
            rear = rear.next;
        }
        rear.next = listNode1 != null ? listNode1 : listNode2;
        return head.next;
    }
}