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