ITPub博客

首页 > 应用开发 > Python > 合并K个排序链表

合并K个排序链表

原创 Python 作者:洗丛显q0 时间:2019-05-15 19:01:02 0 删除 编辑

合并  个排序链表,返回合并后的排序链表。请分析和描述算法的复杂度。

示例:

输入:[
  1->4->5,
  1->3->4,
  2->6
]输出: 1->1->2->3->4->4->5->6
# Definition for singly-linked list.# class ListNode(object):#     def __init__(self, x):#         self.val = x#         self.next = Noneclass Solution(object):
    def mergeKLists(self, lists):
        """
        :type lists: List[ListNode]
        :rtype: ListNode
        """
        #合成一个大的listlist然后排序
        lists = [x for x in lists if x]        if not lists or all([not x for x in lists]): return 
        head = lists.pop()
        
        curr = head        while curr.next:
            curr = curr.next            
        while lists:
            tmp = lists.pop()
            curr.next = tmp            while tmp.next:
                tmp = tmp.next
            curr = tmp        
        if not head or not head.next: return head        return self.mergeSort(head)    
    def mergeSort(self, head):
        if not head.next: return head
        pre, slow, fast = None, head, head        
        while fast and fast.next:
            prev, slow, fast = slow, slow.next, fast.next.next
        
        prev.next = None
        left = self.mergeSort(head)
        right = self.mergeSort(slow)        return self.merge(left, right)    
    def merge(self, left, right):
        if not left:            return right        if not right:            return left        
        if left.val < right.val:
            res = left
            res.next = self.merge(left.next, right)        else:
            res = right
            res.next = self.merge(left, right.next)        return res


来自 “ ITPUB博客 ” ,链接:http://blog.itpub.net/31551893/viewspace-2644529/,如需转载,请注明出处,否则将追究法律责任。

请登录后发表评论 登录
全部评论

注册时间:2019-04-17

  • 博文量
    8
  • 访问量
    4846