题:
https://leetcode.com/problems/merge-k-sorted-lists/
Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.
解法一
将K个链表做K-1次归并,每次归并是对两个链表的归并,最终得到一个排序的链表。即: 1,2合并,遍历2n个节点;(1,2)与3合并,遍历3n个节点。。。(1…k-1)与k合并,遍历kn个节点。所以总共遍历n(2+3+…+k)=n(k^2+k-2)/2,那么时间复杂度为O(n*k^2)。
解法二
对解法一改进一下,改用分治法,所以时间复杂度变为O(nklogk)。
解法三
将K个链表的首元素都取出来,选择出最小的那个作为新链表的head。然后将该元素的next取出来,与其他链表的元素比较再选一个小的,放到新链表中。选择出最小元素的时间复杂度为O(k), 总共要选nk次,所以时间复杂度为O(n*k^2)。
解法四
对解法三改进一下,用最小堆来实现选择最小元素的要求,则时间复杂度降为O(logk),则总的时间复杂度为O(nklogk),与解法二的时间复杂度一样。在Java中,可以使用基于 最小堆算法的PriorityQueue来作为最小堆的集合类。代码如下:
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) {
* val = x;
* next = null;
* }
* }
*/
public class Solution {
public ListNode mergeKLists(List<ListNode> lists) {
if (lists == null || lists.isEmpty()) {
return null;
}
PriorityQueue<ListNode> minHeap = new PriorityQueue<ListNode>(new Comparator<ListNode>(){
public int compare(ListNode l1, ListNode l2) {
return Integer.compare(l1.val, l2.val);
}
});
for (ListNode node : lists) {
if (node != null) {
minHeap.add(node);
}
}
ListNode helper = new ListNode(0);
ListNode next = helper;
while (!minHeap.isEmpty()) {
ListNode min = minHeap.poll(); // must not be null
next.next = min;
min = min.next;
if (min != null) {
minHeap.add(min);
}
next = next.next;
}
return helper.next;
}
}
作者:ywheel
本文出处:http://blog.ywheel.com/post/2015/03/12/leetcode_23/
文章版权归本人所有,欢迎转载,但必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利。