一、题目
合并 k 个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。
示例:
输入:
[
1->4->5,
1->3->4,
2->6
]
输出:
1->1->2->3->4->4->5->6
二、思路
- 非空判断。
- 二层递归。第一层,对list[]集合进行递归,实现遍历。第二层,对任意两个ListNode进行遍历,实现两个ListNode的合并。
- 第一层递归。采用取中点分割输入的方式,然后对输入数组中点两侧的元素分而治之。
- 第二层递归。采用约束函数,依据两个输入结点当前值的大小,对两个输入结点分而治之。递归过程中利用了链表结点的树形结构特征。
三、代码
public class Q023_mergeKLists2 {
public ListNode mergeKLists(ListNode[] lists){
if(lists == null || lists.length == 0) return null;
return merge(lists, 0, lists.length - 1);
}
public ListNode merge(ListNode[] lists, int left, int right){
if(left == right) return lists[left];
//取中点分割输入,很实用的递归方法
int mid = left + (right - left)/2;
ListNode l1 = merge(lists, left, mid);
ListNode l2 = merge(lists, mid + 1, right);
return mergeTwoLists(l1, l2);
}
public ListNode mergeTwoLists(ListNode l1, ListNode l2){
//除非是两者起始都是空的,否则肯定会有一个先是空的
if(l1 == null) return l2;
if(l2 == null) return l1;
//剪枝函数
if(l1.val < l2.val){
l1.next = mergeTwoLists(l1.next, l2);
return l1;
}
else{
l2.next = mergeTwoLists(l1, l2.next);
return l2;
}
}
}