温馨提示×

python归并排序代码怎么写

小亿
84
2023-11-24 23:44:46
栏目: 编程语言

下面是一个用Python实现归并排序的示例代码:

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

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.extend(left[i:])
    result.extend(right[j:])
    
    return result

# 示例用法
arr = [5, 3, 8, 6, 2, 7, 1, 4]
sorted_arr = merge_sort(arr)
print(sorted_arr)

这段代码定义了两个函数:merge_sortmergemerge_sort函数使用递归将数组分成两半,并分别对左右两半进行归并排序。最后,通过调用merge函数将左右两半合并成一个有序数组。merge函数比较左右两个有序数组的元素,将较小的元素添加到结果中,直到其中一个数组的元素全部添加完毕。然后,将剩余的元素添加到结果中。最后,返回合并后的有序数组。在示例用法中,我们定义了一个未排序的数组arr,并使用merge_sort函数对其进行排序,然后打印排序后的结果。

0