为了好玩,我正在尝试实现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)
答案 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]