LeetCode23、合并K个升序链表

LeetCode 系列记录我学习算法的过程。

持续创作，加速成长！这是我参与「掘金日新计划 6 月更文挑战」的第 22 天，点击查看活动详情

题目

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

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

示例：

输入：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


输入： 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

思路

这是我碰到的第一个困难 级别的题目，一开始想用之前那个合并两个有序列表的方法

但是发现实现起来有点困难，就只能使用取巧一点的方法了，将所有节点值取出来排序然后创建一个新链表返回

• 首先判断边界情况，数组为空或只有一项
• 创建一个数组用来存储所有的结点值
• 遍历所有链表，将结点值存入数组
• 将数组升序排序
• 遍历数组，生成新链表并返回

代码实现

/**
 * 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) {
    // 如果传入链表数组为空 返回null
    if(!lists.length) return null
    // 如果只有一个链表，返回改链表
    if(lists.length === 1) return lists[0]
    // 创建一个数组用来存储所有的结点值
    const temp = []
    // 遍历链表数组
    lists.forEach(item => {
        // 遍历当前链表
        while(item) {
            // 将当前结点值存入 temp
            temp.push(item.val)
            // 指向下一结点
            item = item.next
        }
    })
    // 升序排列temp
    temp.sort((a, b) => a - b)
    // 创建头结点
    const res = new ListNode()
    // 创建尾结点
    let last = res
    // 遍历temp
    temp.forEach(t => {
        // 尾结点指向创建的新结点
        last = last.next = new ListNode(t)
    })
    // 返回结果
    return res.next
};

优化

看到一种利用 reduce 的极简写法

• 首先利用 reduce 将所有结点存入一个数组
• 然后通过 sort 进行升序排序
• 最后通过 reduceRight 倒序遍历数组组建新列表返回

var mergeKLists = function(lists) {
    return lists.reduce((p, n) => {
        while (n) {
            p.push(n), n = n.next
        }
        return p
    },[]).sort((a, b) => a.val - b.val).reduceRight((p, n) => (n.next = p, p = n, p), null)
};