leetcode -- Merge k Sorted Lists -- 重点有trick

来源:互联网 发布:普通话转粤语软件 编辑:程序博客网 时间:2024/06/02 02:42

https://leetcode.com/problems/merge-k-sorted-lists/

思路就是把所有的head节点入heap,然后把root,heappop出来。再把被pop出来的node所属于的list的下一个node入heap. 知道heap为空

这里要用到heap,python中有heapq.heapify([])这个函数可以对heap进行create,从而进行操作。

以及 heapq.heappush(), heapq.heappop. 还要注意这里是用tuple(node.val, node)作为heap的元素。

参考http://www.cnblogs.com/zuoyuan/p/3772372.html

0 0
原创粉丝点击