示例:
该题是leetcode的困难题,方案一是和21题一样,先合并两个链表,合并后的链表再和下个链表继续合并,循环下去,这种方案的时间复杂度为O(k^2*n),k表示链表的个数,n表示每个链表的平均节点数。方案一在leetcode上提交不过,即使答案正确。方案二:采用分治的方法,对k个链表进行分治,两两进行合并,时间复杂度O(knlogk),提交通过。
方案二代码:
class Solution {
public ListNode mergeKLists(ListNode[] lists) {
if(lists.length == 0){ //如果lists是空链表,直接返回null
return null;
}
if(lists.length == 1){ //如果lists只有一个链表,返回该链表
return lists[0];
}
int mid = lists.length / 2; //分治的临界点
ListNode[] sub1_list = new ListNode[mid];
ListNode[] sub2_list = new ListNode[lists.length - mid];
for(int i = 0; i < mid; i++){
sub1_list[i] = lists[i]; //存放mid前的链表
}
for(int i = 0; i < lists.length - mid; i++){
sub2_list[i] = lists[mid + i]; //存放mid后的链表
}
//继续分治
ListNode l1 = mergeKLists(sub1_list);
ListNode l2 = mergeKLists(sub2_list);
//合并两个链表
return mergeTwoList(l1, l2);
}
//和21题一样
public static ListNode mergeTwoList(ListNode l1, ListNode l2){
ListNode head = new ListNode();
ListNode pre = head;
while(l1 != null && l2 != null){
if(l1.val < l2.val){
pre.next = l1;
l1 = l1.next;
}else{
pre.next = l2;
l2 = l2.next;
}
pre = pre.next;
}
if(l1 != null){
pre.next = l1;
}
if(l2 != null){
pre.next = l2;
}
return head.next;
}
}