heapsort use heapq; add heap leetcode merge-k-sorted-lists

Author: PegasusWang

docs/15_堆与堆排序/heap_and_heapsort.md

@@ -122,7 +122,7 @@ def test_heapsort_reverse():
 ```
 
 # python 里的 heapq
-python 其实自带了 heapq 模块，用来实现堆的相关操作，原理是类似的。请你阅读相关文档。
+python 其实自带了 heapq 模块，用来实现堆的相关操作，原理是类似的。请你阅读相关文档并使用内置的 heapq 模块完成堆排序。
 
 # 练习题
 
@@ -136,3 +136,7 @@ python 其实自带了 heapq 模块，用来实现堆的相关操作，原理是
 - 《算法导论》第 6 章 Heapsort
 - 《Data Structures and Algorithms in Python》 13.5 节 Heapsort
 - 阅读 Python heapq 模块的文档
+
+# Leetcode
+
+合并 k 个有序链表 https://leetcode.com/problems/merge-k-sorted-lists/description/

docs/15_堆与堆排序/heap_and_heapsort.py

@@ -123,3 +123,18 @@ def test_heapsort_reverse():
     l = list(range(10))
     random.shuffle(l)
     assert heapsort_reverse(l) == sorted(l, reverse=True)
+
+
+def heapsort_use_heapq(iterable):
+    from heapq import heappush, heappop
+    items = []
+    for value in iterable:
+        heappush(items, value)
+    return [heappop(items) for i in range(len(items))]
+
+
+def test_heapsort_use_heapq():
+    import random
+    l = list(range(10))
+    random.shuffle(l)
+    assert heapsort_use_heapq(l) == sorted(l)