[路飞]_23. 合并K个升序链表

题目路径

题目描述

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

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

示例 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 = [[]]
输出： []

解题思路

• 利用归并排序
• 左边处理一下，得到左边的信息
• 右边处理一下，得到右边的信息
• 最后再处理，横跨左右两边的信息

代码

// 归并排序的模板题
var mergeKLists = function(lists) {
    return merge(lists, 0, lists.length - 1);
};

var merge = function(lists, l, r) {
    if(l == r) return lists[l];
    if(l > r) return null;
    // 获取中间值
    let mid = (l + r) >> 1;
    // 获取左半部分，获取右半部分
    return mergeTwoLists(merge(lists, l, mid), merge(lists, mid + 1, r));
}
// 然后做合并
var mergeTwoLists = function(a, b) {
    if (a == null || b == null) return a == null ? b : a;
    let head = new ListNode(0);
    let tail = head, i = a, j = b;
    while(i != null && j != null) {
        if (i.val < j.val) {
            tail.next = i;
            i = i.next;
        } else {
            tail.next = j;
            j = j.next;
        }
        tail = tail.next;
    }
    tail.next = i == null ? j : i;
    return head.next;
}