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

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

你可能感兴趣的:(LeetCode)