数据结构
网络
关系型数据库管理系统
操作系统
Java
MS Excel
iOS
HTML
CSS
Android
Python
C 语言编程
C++
C#
MongoDB
MySQL
Javascript
PHP
物理学
化学
生物学
数学
英语
经济学
心理学
社会学
时尚研究
法律研究使用 Python 中的 heapq 模块合并两个已排序数组？
在本节中，我们将了解如何使用 Python 中的 heapq 模块合并两个已排序列表。例如，如果 list1 = [10, 20, 30, 40] 并且 list2 = [100, 200, 300, 400, 500]，则合并后将返回 list3 = [10, 20, 30, 40, 100, 200, 300, 400, 500]。
为了执行此任务，我们将使用 heapq 模块。此模块作为标准库模块随 Python 一起提供。因此，我们需要在使用它之前导入它。
import heapq
heapq 模块具有一些属性。它们如下所示：
方法 heapq.heapify(iterable)
它用于将可迭代数据集转换为堆数据结构。
方法 heapq.heappush(heap, element)
此方法用于将元素插入堆中。之后重新堆化整个堆结构。
方法 heapq.heappop(heap)
此方法用于返回并从堆顶删除元素，并在其余元素上执行堆化。
方法 heapq.heappushpop(heap, element)
此方法用于在一个语句中插入和弹出元素。
方法 heapq.heapreplace(heap, element)
此方法用于在一个语句中插入和弹出元素。它从堆的根节点删除元素，然后将元素插入堆中。
方法 heapq.nlargest(n, iterable, key=None)
此方法用于从堆中返回 n 个最大元素。
方法 heapq.nsmallest(n, iterable, key=None)
此方法用于从堆中返回 n 个最小元素。
示例代码
import heapq
first_list = [45, 12, 63, 95, 74, 21, 20, 15, 36]
second_list = [42, 13, 69, 54, 15]
first_list = sorted(first_list)
second_list = sorted(second_list)
print('First sorted list: ' + str(first_list))
print('Second sorted list: ' + str(second_list))
final_list = list(heapq.merge(first_list, second_list))
print('The final list: ' + str(final_list))输出
First sorted list: [12, 15, 20, 21, 36, 45, 63, 74, 95] Second sorted list: [13, 15, 42, 54, 69] The final list: [12, 13, 15, 15, 20, 21, 36, 42, 45, 54, 63, 69, 74, 95]
更新于: 2020-06-26
444 次浏览
广告