将三个列表合并在一起

Question

为了好玩，我正在尝试实现一个三路归并排序，也就是k=3。我在递归调用归并排序时没有问题，但在合并三个列表时却得不到一个排序好的列表。基本的想法是，我比较每个列表的第一个元素，如果哪个最小，就把它加到一个新的列表里。我会对所有的数组重复这个过程。

def three_merge(a,b,c):
    i =0
    j =0
    k=0
    list = []
    while(i < len(a) or j < len(b) or k < len(c)):
        while(a[i] <= b[j] and a[i] <= c[k]):
            list.append(a[i])
            i=i+1
            print i
        while(b[j] <= a[i] and b[j] <= c[k]):
            list.append(b[j])
            j=j+1
            print j
        while(c[k] <= a[i] and c[k] <= b[j]):
            list.append(c[k])
            k=k+1
            print k
   return list    

a = [1,2]
b = [-5,10]
c = [-11, 100]
print three_merge(a,b,c)                            

Answer 1

你为什么不把合并三个列表看作是先合并两个列表，然后再把结果和第三个列表合并呢？

def merge(a,b):
    L = a
    l = b
    i = 0
    j = 0
    res = []
    if (len(b)>len(a)):
        L=b
        l=a
    #print L
    #print l
    while (i<len(L)):
        if(len(l)>0):
            while (j<len(l)):
                if (L[i]<=l[j]):
                    res.append(L[i])
                    del L[i]
                    break
                else:
                    res.append(l[j])
                    del l[j]
        else:
            #print res
            res = list(res+L)
            L=[]
            break
    if len(l)>0:
        res = list(res+l)
        l=[]
    return res

def threeMerge(a,b,c):
    return merge(merge(a,b),c)

aa = [1,2]
bb = [-5,3,20]
cc= [-11, 23,45]
print threeMerge(aa,bb,cc)

Answer 2

这里有一个简单的函数，用于多路合并。如果你有任何问题，随时可以问我：

def merge(*args):
    lst = list(args)
    idx = [0] * len(lst)
    out = []

    while lst:
        m = 0
        for i in range(len(lst)):
            if lst[i][idx[i]] < lst[m][idx[m]]:
                m = i
        out.append(lst[m][idx[m]])
        idx[m] += 1
        if idx[m] >= len(lst[m]):
            del lst[m]
            del idx[m]

    return out


print merge([1,2,11], [2,9], [8]) # [1, 2, 2, 8, 9, 11]

Answer 3

问题在于，你在把每个值添加到排序好的列表时，都在增加索引 i, j, k。这就导致在从输入列表中取出所有元素后，索引比对应的输入列表的长度多了一位。所以，当你比较不同输入列表中的元素值时，最终会遇到一个情况，就是你试图去访问 a[len(a)]，这总是会导致索引错误。