合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。
示例:
输入:
[
1->4->5,
1->3->4,
2->6
]
输出: 1->1->2->3->4->4->5->6
思路:
分治法
简单来说就是不停的对半划分,比如k个链表先划分为合并两个k/2个链表的任务,再不停的往下划分,直到划分成只有一个或两个链表的任务,开始合并。举个例子来说比如合并6个链表,那么按照分治法,我们首先分别合并1和4,2和5,3和6。这样下一次只需合并3个链表,我们再合并1和3,最后和2合并就可以了。
public ListNode mergeKLists(ListNode[] lists) {
if(lists == null || lists.length == 0) return null;
return Msort(lists, 0, lists.length - 1);
}
public static ListNode Msort(ListNode[] lists, int low, int high){
if(low < high){
int mid = low + (high - low) / 2;
ListNode leftList = Msort(lists, low, mid);
ListNode righList = Msort(lists, mid + 1, high);
return mergeTwoList(leftList, rightList);
}
return lists[low]; // 如果相等,只有一个元素,返回即可
}
public static ListNode mergeTwoList(ListNode l1, ListNode l2){
ListNode res = null;
if(l1 == null) return l2;
if(l2 == null) return l1;
if(l1.val < l2.val){
res = l1;
res.next = mergeTwoList(l1.next, l2);
}else{
res = l2;
res.next = mergeTwoList(l1, l2.next);
}
return res;
}