2022.04.11 合并K个升序链表

题目地址(23. 合并K个升序链表)

https://leetcode-cn.com/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

前置知识

• 链表

思路

• 合并两个链表的进阶。把两两合并封装成一个函数，在链表数组里，两两合并即可。

代码

• 语言支持：JavaScript

JavaScript Code:

/**
 * Definition for singly-linked list.
 * function ListNode(val, next) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.next = (next===undefined ? null : next)
 * }
 */
/**
 * @param {ListNode[]} lists
 * @return {ListNode}
 */
var mergeKLists = function(lists) {
    let res = null;
    for (let i = 0; i < lists.length; i++) {
        res = merge (res, lists[i]);
    }
    return res;
};

function merge (list1, list2) {
    let pre = new ListNode(0);
    let next = pre;
    if (list1 == null) {
        return list2;
    } else if (list2 == null) {
        return list1;
    }
    while (list1 !== null && list2 !== null) {
        if (list1.val >= list2.val) {
            pre.next = list2;
            list2 = list2.next;
            pre = pre.next;
        } else {
            pre.next = list1;
            pre = list1;
            list1 = list1.next;
        }
    }
    if (list1 == null) {
        pre.next = list2;
    } else if (list2 == null) {
        pre.next = list1;
    }
    return next.next;
}

复杂度分析

• 时间复杂度：$O(n)$ 第一次合并n，第2次合并2 * n，第k次合并k * n。加起来就是 (k * (1 + k) / 2) * n
• 空间复杂度：$O(1)$