题目难度:
Hard
题目要求:
给你一个链表数组,每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中,返回合并后的链表。
示例 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
算法思路:
1.先考虑设计一个函数,合并两个有序链表:
(1)需要一个head指针,用于最后返回合并后的链表,它一开始可以指向一个0 的节点,最后返回head.next;
(2)需要aPtr和bPtr,分别指向两个链表还没有合并部分的第一个节点;
(3)需要一个tail指针,指向已经合并好的部分的最后一个节点;
2.当链表a或者链表b一开始就为空时,直接返回非空的那个链表;
3.按照思路1设计好之后,比较aPtr和bPtr指向的值的大小,如果aptr更小,就将aptr的那个节点接到tail后面,tail后移一位,aptr也后移一位;
4.其他情况就将bptr的节点接到tail后面,tail后移一位,aptr也后移一位;
5.当aptr或者bptr为空时,就将剩下非空的那个指针接到tail后面,合并成一个链表,最后返回head.next就可以。
6.完成两个链表的合并后,得到一个链表ans,然后又将ans与第三个链表合并,循环直至合并成一个链表。
代码:
/*
* @lc app=leetcode.cn id=23 lang=java
*
* [23] 合并K个升序链表
*/
// @lc code=start
/**
* 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 ans=null;
for(int i=0;i<lists.length;i++){
ans=mergeTwoLists(ans, lists[i]);
}
return ans;
}
public ListNode mergeTwoLists(ListNode a,ListNode b){
if(a==null ||b == null){
return a!=null?a:b;
}
ListNode head = new ListNode(0);
ListNode tail = head;
ListNode aPtr = a;
ListNode bPtr = b;
while(aPtr!=null && bPtr!=null){
if(aPtr.val<bPtr.val){
tail.next= aPtr;
aPtr=aPtr.next;
}else{
tail.next= bPtr;
bPtr=bPtr.next;
}
tail =tail.next;
}
tail.next=(aPtr==null?bPtr:aPtr);
return head.next;
}
}
时间复杂度:渐进时间复杂度为 O(k^2 n)
空间复杂度:O(1)
【没搞懂为什么空间复杂度是O(1),有理解的朋友还麻烦评论解答一下,谢谢!!!】