Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.
Example:
Input:
[
1->4->5,
1->3->4,
2->6
]
Output: 1->1->2->3->4->4->5->6
题目的意思:输入一个有序列表数组,将他们合并成一个升序的列表输出。
这题最大的难点在于很容易爆内存,如果只是简单粗暴地遍历每个列表,依次比较大小再排序。
所以参考了大佬的做法,用了比较稳定的复杂度为O(nlogn)的归并排序,将大问题分治来处理,换而言之也就是递归。问题最小细化到两个列表排序后输出,也就跟之前做过的一题一样了。
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode mergeKLists(ListNode[] lists) {
if(lists.length == 0 || lists == null){
return null;
}
return Merge(lists,0,lists.length-1);
}
ListNode Merge(ListNode[] lists,int l,int r){
if(l<r){
int m=(l+r)/2;
return MergeSort(Merge(lists,l,m),Merge(lists,m+1,r));//递归直到处理的为两个列表
}
return lists[l];//返回是l或r结果一样
}
ListNode MergeSort(ListNode l1,ListNode l2){
ListNode headNode=new ListNode(0);
ListNode cur=headNode;
while(l1 != null && l2 != null){
if(l1.val<l2.val){
cur.next=l1;
l1=l1.next;
}
else{
cur.next=l2;
l2=l2.next;
}
cur=cur.next;
}
if(l1 != null){
cur.next=l1;
}
else if(l2 != null){
cur.next=l2;
}
return headNode.next;
}
}