在python中优化合并排序功能

Question

有什么想法可以优化此合并排序功能？

输入是一个像这样的列表：[[（（1，'i'），（3，'i'），（5，'i'），（8，'i'）]]，[（2，'n' ）]，[（4，'t'），（7，'t'）]，[（6，'a'）]，[（9，'v'）]，[（10，'e'）] ]，然后输出单词：“ initiative”

def merge(decks):
    while len(decks) > 1:
        del1 = decks.pop(0)
        del2 = decks.pop(0)
        total = list()
        while (len(del1) and len(del2)) > 0:
            if del1[0] < del2[0]:
                total.append(del1.pop(0))
            else:
                total.append(del2.pop(0))
        total.extend(del1)
        total.extend(del2)
        decks.append(total)

    word = ""
    for kort in decks[0]:
        word += kort[1]
    return word

Answer 1

我可以想到一个优化：在内部循环之前反转del[0]和del[1]并将list.pop(0) （ 一个O（n）操作 ）更改为list.pop() （在O中（1））。 反转也是O（n），但是由于您在循环内执行pop(0) ，所以实际上是O(n^2) 。

Answer 2

简单得多就是简单地将列表弄平，然后自然地对元组进行排序。

from itertools import chain, imap
from operator import itemgetter 

def merge(decks):
    return "".join(imap(itemgetter(1),
                   sorted(chain.from_iterable(decks)))

• from_iterable平嵌套列表
• sorted每个元组的整数first元素自然地对展平序列中的元组sorted排序
• itemgetter(1)是lambda x: x[1]的有效实现lambda x: x[1]
• imap将访问功能应用于sorted返回的列表，提取字母
• "".join从提取的字母中构建一个新的字符串

风格上“功能性”稍差一点的是使用列表理解，这很有意义，因为sorted无论如何都会返回一个列表，而不是更通用的迭代器。

def merge(decks):
    return "".join(x[1]
                   for x in sorted(chain.from_iterable(decks)))

您还可以使用heapq模块的merge函数，该函数利用了每个嵌套列表都已排序的事实，而不是简单地对扁平化列表进行排序。

import heapq
def merge(decks):
    return "".join(x[1] for x in heapq.merge(*decks))