【力扣刷题】23. 合并K个升序链表

109 阅读1分钟

Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情

一、题目描述

来源:力扣(LeetCode)

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

请你将所有链表合并到一个升序链表中,返回合并后的链表。   示例 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

二、思路分析

  1. 创建一个根节点res,值为0
  2. 创建一个list保存所有链表的数组
  3. 然后使用stream流排序
  4. 按顺序给添加节点,最后去除根节点返回就是一个合并后的有序链表

三、代码实现

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
          //新建一个头节点
        ListNode res = new ListNode(0);
        //当前节点指针
        ListNode cur = res;
        List<Integer> list = new ArrayList<>();
        for (ListNode listNode : lists) {
            while (listNode != null) {
                list.add(listNode.val);
                listNode = listNode.next;
            }
        }
        List<Integer> collect = list.stream().sorted().collect(Collectors.toList());
        for (int i = 0; i < collect.size(); i++) {
            ListNode listNode = new ListNode(collect.get(i));
            cur.next = listNode;
            cur = cur.next;
        }
        return res.next;
    }
}

四、运行结果

image.png

总结

这道题跟 21. 合并两个有序链表 一样的解法,只是从两个链表变为多个