快速排序 python
时间: 2023-07-15 14:08:15 浏览: 48
快速排序(Quick Sort)是一种常用的排序算法,它的思想是选择一个基准元素,将数组分为两部分,使得左边的元素都小于等于基准元素,右边的元素都大于等于基准元素,然后对左右两部分递归地进行排序。
在Python中,可以使用以下代码实现快速排序:
```python
def quick_sort(arr):
if len(arr) <= 1:
return arr
else:
pivot = arr[0]
less = [x for x in arr[1:] if x <= pivot]
greater = [x for x in arr[1:] if x > pivot]
return quick_sort(less) + [pivot] + quick_sort(greater)
```
这个函数接受一个数组作为输入,并返回一个排序后的新数组。它首先检查数组的长度,如果长度小于等于1,则直接返回数组本身。否则,选择第一个元素作为基准元素(pivot),然后用列表推导式将比基准元素小的元素放入`less`列表,将比基准元素大的元素放入`greater`列表。最后,使用递归对`less`和`greater`进行排序,并将排序后的结果与基准元素合并返回。
你可以这样调用函数来测试快速排序的实现:
```python
arr = [9, 3, 7, 5, 1, 8, 2, 6, 4]
sorted_arr = quick_sort(arr)
print(sorted_arr)
```
输出结果为:`[1, 2, 3, 4, 5, 6, 7, 8, 9]`,表示数组已成功排序。