欢迎访问宙启技术站
智能推送

用Python编写归并排序算法

发布时间:2023-06-21 07:25:29

归并排序是一种基于分治思想的排序算法,在实践中得到广泛应用。它的基本思想是将一个大的问题分成若干个小问题,通过递归求解小问题,最后合并得到整个问题的解决方案。归并排序算法的时间复杂度为O(nlogn),它具有稳定性,能够对任何类型的数据进行排序。下面我们将介绍如何用Python编写归并排序算法。

1.实现归并函数

归并过程是将两个有序数组合并成一个有序数组。我们可以通过两个指针分别遍历两个数组,然后比较大小,将较小的元素放入新的数组中。代码如下:

def merge(left, right):
    result = []
    i = j = 0
    while i < len(left) and j < len(right):
        if left[i] <= right[j]:
            result.append(left[i])
            i += 1
        else:
            result.append(right[j])
            j += 1
    result += left[i:]
    result += right[j:]
    return result

2.实现归并排序函数

归并排序算法是基于分治思想的,它将一个序列分成两个子序列,然后将子序列递归地排序,最后将两个子序列合并成一个有序序列。整个过程可以看作是一个递归的过程,递归结束的条件是子序列的长度小于等于1。代码如下:

def merge_sort(arr):
    if len(arr) <= 1:
        return arr
    mid = len(arr) // 2
    left = merge_sort(arr[:mid])
    right = merge_sort(arr[mid:])
    return merge(left, right)

3.测试归并排序函数

最后,我们编写一个测试函数,随机生成一个长度为10的数组,然后调用归并排序函数对数组进行排序,最后输出排序结果。代码如下:

import random

def test_merge_sort():
    arr = [random.randint(1, 100) for i in range(10)]
    print("Before sorting:", arr)
    sorted_arr = merge_sort(arr)
    print("After sorting:", sorted_arr)

if __name__ == '__main__':
    test_merge_sort()

以上就是用Python编写归并排序算法的全部代码和详细解释,希望对大家学习和应用归并排序算法有所帮助。