python如何使用merge实现堆

736次阅读
没有评论

python如何使用merge实现堆

python merge说明

对于较大的数据集,将会占用大量内存。不是对整个组合序列进行排序,而是使用 merge() 一次生成一个新序列。

python merge堆实例

import heapq
import random
 
 
random.seed(2016)
 
data = []
for i in range(4):
    new_data = list(random.sample(range(1, 101), 5))
    new_data.sort()
    data.append(new_data)
 
for i, d in enumerate(data):
    print('{}: {}'.format(i, d))
 
print('\nMerged:')
for i in heapq.merge(*data):
    print(i, end=' ')
print()
 
# output
# 0: [33, 58, 71, 88, 95]
# 1: [10, 11, 17, 38, 91]
# 2: [13, 18, 39, 61, 63]
# 3: [20, 27, 31, 42, 45]
#
# Merged:
# 10 11 13 17 18 20 27 31 33 38 39 42 45 58 61 63 71 88 91 95

因为merge()使用堆的实现,它根据被合并的序列元素个数消耗内存,而不是所有序列中的元素个数。

神龙|纯净稳定代理IP免费测试>>>>>>>>天启|企业级代理IP免费测试>>>>>>>>IPIPGO|全球住宅代理IP免费测试

相关文章:

版权声明:wuyou2021-04-29发表,共计612字。
新手QQ群:570568346,欢迎进群讨论 Python51学习