美文网首页
Leetcode-23题:Merge k Sorted List

Leetcode-23题:Merge k Sorted List

作者: 八刀一闪 | 来源:发表于2016-09-24 16:52 被阅读63次

    题目:

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

    思路:
      这个题首先想到的是与合并两个有序链表一样,维护k个指针分别指向那k个链表,每次从中选出一个最小的加入到已有序链表尾,然后那个指针后移。这种做法的时间复杂度为o(nk),提交之后,结果超时。然后想到当某个指针为空时,便不再需要保存这个指针,修改代码为动态删除空指针也超时,因为本质上时间复杂度还是o(nk)。合并k个链表其实可以分解为分别合并前一半和后一半,然后再将这两个结果合并,也就是分治的思想。采用分治的话,层次为log(n),每次合并都为o(n),所以最后的时间复杂度为o(nlog(n))。

    代码:

    # Definition for singly-linked list.
    # class ListNode(object):
    #     def __init__(self, x):
    #         self.val = x
    #         self.next = None
    
    class Solution(object):
    
        def mergeTwo(self, l1, l2):
            head = ListNode(None)
            p = head
            while l1!=None and l2!=None:
                p.next = l1
                if l1.val < l2.val:
                    p = p.next
                    l1 = l1.next
                else:
                    p.next = l2
                    p = p.next
                    l2 = l2.next
            if l1 != None:
                p.next = l1
            if l2 != None:
                p.next = l2
            return head.next
    
        def mergeKLists(self, lists):
            """
            :type lists: List[ListNode]
            :rtype: ListNode
            """
            if lists==None or len(lists)==0:
                return None
            if len(lists) == 1:
                return lists[0]
            l = self.mergeKLists(lists[0:len(lists)/2])
            r = self.mergeKLists(lists[len(lists)/2:])
            return self.mergeTwo(l,r)
    
    

    相关文章

      网友评论

          本文标题:Leetcode-23题:Merge k Sorted List

          本文链接:https://www.haomeiwen.com/subject/vtvxyttx.html