No.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 = [[]]
输出：[]

提示：

• 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 个升序链表，和 #88 很像，但有两点区别：

• 此题数据结构为链表，由于是线性的，相对不复杂
• 此题需合并 k 个元素

实现

分冶

/**
 * 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) {
  // 拆分
  function merge(l1, l2) {
    const head = new ListNode();
    let current = head;
    while (l1 && l2) {
      if (l1.val < l2.val) {
        current.next = l1;
        l1 = l1.next;
      } else {
        current.next = l2;
        l2 = l2.next;
      }
      current = current.next;
    }

    current.next = l1 || l2;

    return head.next;
  }

  // 分冶，归并排序
  function dfs(i, j) {
    const len = j - i;
    if (!len) return null;
    if (len === 1) return list[i];

    const mid = len >> 1;
    const l1 = dfs(i, i + mid); // 合并左半部分
    const l2 = dfs(i + mid, j); // 合并右半部分
    return merge(l1, l2);
  }

  // 合并
  return dfs(0, lists.length);
};