Python中的合并排序实现

Question

def merge_sort(arr):
    arr1 = []
    arr2 = []
    mid = int(len(arr)/2)
    if len(arr) == 1:
        return arr
    elif len(arr) > 2:
        arr1 = arr[0:mid]
        arr2 = arr[mid:len(arr)]
        return [merge_sort(arr1), merge_sort(arr2)]
    elif len(arr) == 2:
        if arr[0]>arr[1]:
            temp = arr[0]
            arr[0] = arr[1]
            arr[1] = temp
        return arr

我写过这个函数，它将列表递归地分成两半，直到它的长度变为至少为1或2.

如果列表的长度变为：

1. 如果列表的长度变为
，则按原样返回列表
2. 交换列表的元素，使它们按升序排列，并返回列表。

问题在于，在此代码运行之后，输出以列表列表的形式出现，并且合并它们似乎是一个凝灰岩作业。

代码为给定的调用生成以下输出：

merge_sort([5,4,1,8,7,2,6,3,9])

输出：

[[[4, 5], [1, 8]], [[2, 7], [[6], [3, 9]]]]

Answer 1

在Merge Sort Algorithm中，您必须具有merge功能，以便在合并的每个步骤中，您可以对两个小部分进行排序。问题是，当数组大小为2时，您只是交换两个数字。

def merge(arr1,arr2):
    it1 = 0
    it2 = 0
    ret = []
    while it1 < len(arr1) or it2 < len(arr2):
        if it1 == len(arr1):
            ret.append(arr2[it2])
            it2+=1
        elif it2 == len(arr2):
            ret.append(arr1[it1])
            it1+=1
        else:
            if arr1[it1] > arr2[it2]:
                ret.append(arr2[it2])
                it2+=1
            else:
                ret.append(arr1[it1])
                it1+=1
    return ret



def merge_sort(arr):
    arr1 = []
    arr2 = []
    mid = int(len(arr)/2)
    if len(arr) == 1:
        return arr
    elif len(arr) > 2:
        arr1 = arr[0:mid]
        arr2 = arr[mid:len(arr)]
        return merge(merge_sort(arr1), merge_sort(arr2))
    elif len(arr) == 2:
        if arr[0]>arr[1]:
            temp = arr[0]
            arr[0] = arr[1]
            arr[1] = temp
    return arr

PS。我刚刚编辑了一些小的修改代码。您可以更好地编写合并排序算法，并且更加整洁。