我正在尝试使用python中的递归函数为合并排序算法实现python代码。但是代码无法按预期工作。代码中的q值未如预期的那样减小,而是停留在第1个值上,并且该函数属于无限递归函数。
def merge(A,p,q,r):
""" This function merges 2 sorted subarrays A[p .. q] and A[q+1 .. r] into a single sorted array A[p .. r]"""
A1 = A[p:q]
A2 = A[q:r]
A1 = A1 + [max(A)+1]
A2 = A2 + [max(A)+1]
#print(A1,A2)
i, j = 0,0
for k in range(p,r):
if(A1[i]<=A2[j]):
A[k] = A1[i]
i+=1
else:
A[k] = A2[j]
j+=1
#print(i,j,k)
#print(A)
return A
def mrgSort(A,p,r):
r=len(A)
if(p<(r-1)):
q = (p+r)//2
print(q)
A1 = mrgSort(A,p,q)
#print(A)
A2 = mrgSort(A,q,r)
#print(A)
A = [A1,A2]
merge(A,p,q,r)
#print(A)
return A
print(mrgSort([0,3,2,-1,9,5,6,2,1],0,9))
#print(merge([0,3,5,10,-1,1,5,7,9],0,4,9))
答案 0 :(得分:0)
def merge(A,p,q,r):
""" This function merges 2 sorted subarrays A[p .. q] and A[q+1 .. r] into a single sorted array A[p .. r]"""
A1 = A[p:q]
A2 = A[ q:r]
A1 = A1 + [(max(A)+1)]
A2 = A2 + [(max(A)+1)]
i, j = 0,0
for k in range(p,r):
if A1[i]<=A2[j]:
A[k] = A1[i]
i+=1
else:
A[k] = A2[j]
j+=1
return A
def mrgSort(A,p,r):
if(p<(r-1)):
q = (p+r)//2
A = mrgSort(A,p,q)
A = mrgSort(A,q,r)
A = merge(A,p,q,r)
return A
您犯的一些错误:
每次执行r=len(A)
都会重置变量r。
A = [A1,A2]为您提供列表列表。如果要串联列表,请执行以下操作:A = A1 + A2
当您查看算法的实现方式时,您对A1和A2的使用是错误的。