kingforce 发表于 2015-12-3 12:30:51

leetcode 【 Merge k Sorted Lists 】python 实现

  题目:
  Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.
  
  代码:oj测试通过 Runtime: 231 ms



1 # Definition for singly-linked list.
2 # class ListNode:
3 #   def __init__(self, x):
4 #         self.val = x
5 #         self.next = None
6
7 class Solution:
8   # @param a list of ListNode
9   # @return a ListNode
10   def mergeKLists(self, lists):
11         # none
12         if len(lists) == 0:
13             return None
14         # only one list
15         if len(lists) == 1:
16             return lists
17         # merge sort each two linked list
18         l1 = self.mergeKLists(lists[:len(lists)/2])
19         l2 = self.mergeKLists(lists)
20         head = self.mergeTwoLists(l1,l2)
21         return head
22            
23   # merge two sorted linked list   
24   def mergeTwoLists(self, l1, l2):
25         if l1 is None:
26             return l2
27         if l2 is None:
28             return l1
29         p = ListNode(0)
30         dummyhead = p
31         while l1 is not None and l2 is not None:
32             if l1.val < l2.val:
33               p.next = l1
34               l1 = l1.next
35               p = p.next
36             else:
37               p.next = l2
38               l2 = l2.next
39               p = p.next
40         if l1 is None:
41             p.next = l2
42         else:
43             p.next = l1
44         return dummyhead.next
  思路:
  总体的思路是模仿归并排序。
  这里给的参数是多个链表的头结点数组,将头节点数组不断二分;直到只剩下一个头结点,返回该头节点到上一层,并在上一层中将两个有序链表合并。
  将两个有序链表merge的代码和思路,在这篇日志中可以看到
  http://www.cnblogs.com/xbf9xbf/p/4186905.html
  另,网上还有一种用堆的方法,后续再去试探这种方法,并把另一种方法的代码补上。
  另,关于该问题的算法复杂度O(nklogk) (n为单个链表的最大长度,k为链表个数),个人觉得下面这篇日志表述的比较好,共享一下
  http://www.tuicool.com/articles/ZnuEVfJ
页: [1]
查看完整版本: leetcode 【 Merge k Sorted Lists 】python 实现