首页 > 代码库 > 【LeetCode】Merge k Sorted Lists

【LeetCode】Merge k Sorted Lists

题目

Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.

解答

方法1:利用分治的思想把合并k个链表分成两个合并k/2个链表的任务,一直划分,知道任务中只剩一个链表或者两个链表。可以很简单的用递归来实现。因此算法复杂度为T(k) = 2T(k/2) + O(nk),很简单可以推导得到算法复杂度为O(nklogk)

方法2:先链表1、2合并,12结果和3合并,123结果和4合并,…,123..k-1结果和k合并,merge sort,外面套一层循环即可。超时。时间复杂度是O(n*(k^2+k-2)/2) = O(nk^2),参考:http://www.cnblogs.com/TenosDoIt/p/3673188.html

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) {
 *         val = x;
 *         next = null;
 *     }
 * }
 */
public ListNode mergeKLists(List<ListNode> lists){
	int n=lists.size();
	if(n==0){
		return null;
	}
	while(n>1){
		int k=(n+1)/2;
		for(int i=0;i<n/2;i++){
			lists.set(i,merge(lists.get(i),lists.get(i+k)));
		}
		n=k;
	}
	return lists.get(0);
}

private ListNode merge(ListNode l1,ListNode l2){
	ListNode fakeHead=new ListNode(-1);
	ListNode l3=fakeHead;                //
	if(l1!=null||l2!=null){
		while(l1!=null&&l2!=null){
			if(l1.val<l2.val){
				l3.next=l1;
				l1=l1.next;
				l3=l3.next;
			}else{
				l3.next=l2;
				l2=l2.next;
				l3=l3.next;
			}
		}
		if(l1!=null){
			l3.next=l1;
		}
		if(l2!=null){
			l3.next=l2;
		}
	}
	return fakeHead.next;
}


//方法2:超时
public ListNode mergeKLists(List<ListNode> lists){
	ListNode fakeHead=new ListNode(-1);
	for(int i=0,len=lists.size();i<len;i++){
		ListNode p1=fakeHead.next;
		ListNode p2=lists.get(i);
		ListNode pre=fakeHead;
		while(p1!=null&&p2!=null){
			if(p1.val<p2.val){
				pre.next=p1;
				p1=p1.next;
			}else{
				pre.next=p2;
				p2=p2.next;
			}
			pre=pre.next;
		}
		if(p1!=null){
			pre.next=p1;
		}
		if(p2!=null){
			pre.next=p2;
		}
	}
	return fakeHead.next;
}

---EOF---

【LeetCode】Merge k Sorted Lists