0

为了好玩,我正在尝试实现 k 方式合并排序,其中 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)                            
4

3 回答 3

1

问题是i, j, k您将附加到排序列表的每个值的索引提前,这意味着在从列表中获取所有元素后,索引比相应输入列表的长度大一。因此,当您比较不同输入列表的元素值时,您最终会达到您想要做的点,实际上a[len(a)],虽然总是会给出索引错误。

于 2013-02-08T09:22:08.650 回答
1

这是一个用于 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]
于 2013-02-08T11:05:30.257 回答
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)
于 2013-02-08T09:35:42.680 回答