使用python写一个归并排序算法
时间: 2023-10-24 15:09:09 浏览: 125
归并排序是一种经典的排序算法,它采用分治策略,将一个大问题分解成多个小问题,然后将小问题解决后合并起来得到原问题的解。下面是使用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 = 0
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
```
这个代码中,我们定义了两个函数:`merge_sort`和`merge`。其中,`merge_sort`函数是归并排序的主体部分,它将一个未排序的数组作为输入,并返回一个已排序的数组。首先,如果输入数组的长度小于等于1,则直接返回该数组。否则,我们找到数组的中间位置,并将其分成左右两个子数组。然后再递归调用`merge_sort`函数来对左右两个子数组进行排序。最后,我们将左右两个有序的子数组合并成一个有序的数组并返回。
`merge`函数用于将两个已排序的子数组合并成一个更大的已排序的数组。它使用两个指针`i`和`j`来分别指向左右两个子数组的第一个元素,并比较它们的大小。如果左子数组的元素小于等于右子数组的元素,则将左子数组的元素添加到结果数组中,并递增`i`指针;否则将右子数组的元素添加到结果数组中,并递增`j`指针。最后将剩余的元素追加到结果数组中。
最后,我们可以测试一下这个归并排序算法:
```
arr = [2, 1, 4, 3, 6, 5]
sorted_arr = merge_sort(arr)
print(sorted_arr)
```
输出结果为:
```
[1, 2, 3, 4, 5, 6]
```
可以看到,这个归并排序算法成功地将输入的未排序数组排序成了一个有序的数组。
阅读全文