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

Python实现归并排序算法

发布时间:2023-12-04 08:58:29

归并排序是一种基于分治法的排序算法,它将待排序的列表不断地分成两个子列表,直到每个子列表只有一个元素,然后再将这些子列表进行合并,最终得到一个有序的列表。

归并排序算法的实现可以分为两个步骤:划分和合并。首先,将待排序的列表划分为较小的子列表,直到每个子列表只有一个元素。接下来,使用归并操作将这些子列表不断地合并,直到全部子列表合并为一个有序的列表。

下面是一个用Python实现归并排序算法的例子:

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)

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

在这个例子中,merge_sort函数接受一个列表作为输入,并返回排序后的列表。如果列表的长度小于等于1,则直接返回该列表。否则,将列表划分为两个子列表,分别递归地调用merge_sort函数来排序这两个子列表,然后再使用merge函数将排序好的子列表合并。

merge函数接受两个有序的子列表作为输入,并返回合并后的有序列表。它使用了两个指针ij来追踪两个子列表的位置,通过比较两个子列表中的元素大小,将较小的元素加入到结果列表merged中,并移动对应的指针。最后,将剩余的元素加入到merged中,然后返回合并后的结果。

下面是一个使用例子:

arr = [6, 5, 3, 1, 8, 7, 2, 4]
sorted_arr = merge_sort(arr)
print(sorted_arr)

运行上述代码,将输出[1, 2, 3, 4, 5, 6, 7, 8],表示归并排序算法成功地对输入的列表进行了排序。

归并排序算法的时间复杂度为O(nlogn),其中n是待排序列表的长度。这使得归并排序成为一种比较高效的排序算法。然而,归并排序算法的缺点是需要使用额外的空间来存储排序的结果,因此在内存有限的情况下可能会存在一定的问题。