归并排序(python)

def merge(A, B):
'''
合并两个已排序的数组

Args:
list A:已排序数组A
list B:已排序数组B

Returns:
list result: 已合并的有序数组
'''
i,j = 0,0
result = []
while i<len(A) and j<len(B):
if A[i]<B[j]:
result.append(A[i])
i += 1
else:
result.append(B[j])
j += 1
result += A[i:]
result += B[j:]
return result

def merge_sort(A):
'''
归并排序
思路:
不断递归, 直到只有一个数值, 归并排序可以不要将一个数与其他所有的额数对照, 减少操作次数
'''
if len(A)==1:
#当最后为叶子节点时候返回数组
return A
else:
m = len(A)//2
#构造左二叉树
left = merge_sort(A[:m])
#构造右二叉树
right = merge_sort(A[m:])
#合并左右
return merge(left, right)

测试:
A = [22,12,45,25,56]
print(merge_sort(A))