合并三个列表

时间:2013-02-08 08:39:06

标签: python merge

为了好玩,我正在尝试实现k方式合并排序,其中k = 3。我没有遇到以递归方式调用mergesort的问题,但我试图将三个列表合并在一起,但我没有得到一个排序列表。基本的想法是我比较每个列表的第一个元素,如果它是最小的,我将它附加到列表中。我重复所有阵列的过程。

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)                            

3 个答案:

答案 0 :(得分:1)

问题是您为追加到排序列表的每个值推进索引i, j, k,这意味着在从列表中获取所有元素之后,索引比相应输入列表的长度大1。因此,当您比较不同输入列表的元素值时,最终会达到您尝试执行的点a[len(a)],同时始终会产生索引错误。

答案 1 :(得分:0)

为什么不考虑将三个列表合并为两个列表的两个合并的组合

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)

答案 2 :(得分:0)

这是一个简单的n路合并功能。如果您有任何疑问,请随时询问:

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]