python算法 ---- 归并排序

归并排序

归并排序是采用分治法的一个非常典型的应用。归并排序的思想就是先递归分解数组，再合并数组。

将数组分解最小之后，然后合并两个有序数组，基本思路是比较两个数组的最前面的数，谁小就先取谁，取了后相应的指针就往后移一位。然后再比较，直至一个数组为空，最后把另一个数组的剩余部分复制过来即可。

基本思想

归并排序是用分治思想，分治模式在每一层递归上有三个步骤：

• 分解（Divide）：将n个元素分成个含n/2个元素的子序列。
• 解决（Conquer）：用合并排序法对两个子序列递归的排序。
• 合并（Combine）：合并两个已排序的子序列已得到排序结果。

具体过程如下图所示：

上图中首先把一个未排序的序列从中间分割成2部分，再把2部分分成4部分，依次分割下去，直到分割成一个一个的数据，再把这些数据两两归并到一起，使之有序，不停的归并，最后成为一个排好序的序列。

时间复杂度

• 最优时间复杂度：O(nlogn)
• 最坏时间复杂度：O(nlogn)
• 稳定性：稳定

代码实现

代码1>

def merge(left_arr, right_arr):
    """合并操作：将两个有序数组合并为一个有序数组"""

    left_index, right_index = 0, 0  # left_index, right_index 分别表示左边列表和右边列表的索引号
    result = []

    # 当左边列表或者右边列表走到头了，循环结束
    while left_index < len(left_arr) and right_index < len(right_arr):
        if left_arr[left_index] < right_arr[right_index]:
            result.append(left_arr[left_index])
            left_index += 1
        else:
            result.append(right_arr[right_index])
            right_index += 1
    result += (left_arr[left_index:])   # 把多余一边的元素给加到 result 列表中
    result += (right_arr[right_index:])
    return result


def merge_sort(arr):
    if len(arr) <= 1:
        return arr

    mid = len(arr) // 2
    left_arr = merge_sort(arr[:mid])
    right_arr = merge_sort(arr[mid:])
    merge_arr = merge(left_arr, right_arr)
    return merge_arr


if __name__ == '__main__':
    numbers = [3, 9, 8, 5, 14, 23, 54, 26, 93, 17, 77, 31, 10, 1]
    print(merge_sort(numbers))

代码2>

def merge(arr, low, high, mid):
    """合并数组"""
    i = low
    j = mid + 1
    result = []
    while i <= mid and j <= high:  # 只要左右两边都有数
        if arr[i] <= arr[j]:
            result.append(arr[i])
            i += 1
        else:
            result.append(arr[j])
            j += 1

    # while执行完，肯9定有一部分没数了
    while i <= mid:
        result.append(arr[i])
        i += 1
    while j <= high:
        result.append(arr[j])
        j += 1
    arr[low:high + 1] = result


def merge_sort(arr, low, high):
    """归并排序"""
    if low == high:
        return
    mid = (low + high) // 2
    merge_sort(arr, low, mid)
    merge_sort(arr, mid + 1, high)
    merge(arr, low, high, mid)


if __name__ == "__main__":
    numbers = [10, 8, 5, 2, 41, 58, 32, 24, 9, 3, 2, 0, 1]
    merge_sort(numbers, 0, len(numbers) - 1)
    print(numbers)

return:
当函数中需要返回一个新的列表时，就需要加上return

比如函数定义的形参列表是 arr, 结果想要得到另一个列表result,这时候就需要使用 return result，并且在主函数中要用一个新的变量来接收函数执行后的结果，如 list = merge_sort(numbers, 0, len(numbers) - 1),print(list)

如果函数没有使用return语句的话，主函数中直接用一个变量来接收函数执行后的结果会返回None

所以，使用 return语句，需要用变量来接收函数执行后的结果，如果没有使用return语句，就不要使用变量来接收函数执行后的结果

归并排序的代码执行过程