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.length0 <= k <= 10^40 <= lists[i].length <= 500-10^4 <= lists[i][j] <= 10^4lists[i]按 升序 排列lists[i].length的总和不超过10^4
二、思路分析
- 创建一个根节点res,值为0
- 创建一个list保存所有链表的数组
- 然后使用stream流排序
- 按顺序给添加节点,最后去除根节点返回就是一个合并后的有序链表
三、代码实现
/**
* 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;
}
}
四、运行结果
总结
这道题跟 21. 合并两个有序链表 一样的解法,只是从两个链表变为多个