首页 > itarticle > leeetcode回顾

leeetcode回顾

admin 1月 29, 2021 0

题目

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

示例:

输入:
[
  1->4->5,
  1->3->4,
  2->6
]
输出: 1->1->2->3->4->4->5->6

题解

这其实就是归并排序从二路变为多路，原来二路的适合，可以直接比较两路得出较小的数字，而现在换成了多路，因此就可以考虑使用小顶堆这个数据结构，每次将k路的开头元素放入堆中；每取出一个最小值后，将该最小值所对应的路的开头元素放入小顶堆中，开始新一轮的选取排序

AC代码

class (object):
    def mergeKLists(self, lists):
        """
        :type lists: List[ListNode]
        :rtype: ListNode
        """
        import heapq
        h = []
        head = ListNode(-1)
        tmp = head
        for i in range(len(lists)):
            if lists[i] != None:
                heapq.heappush(h, (lists[i].val, i))
        while len(h) != 0:
            _, loc = heapq.heappop(h)
            tmp.next = lists[loc]
            tmp = tmp.next
            lists[loc] = lists[loc].next
            if lists[loc] != None:
                heapq.heappush(h, (lists[loc].val, loc))
        return head.next