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

Python代码编写案例:如何实现快速排序算法

发布时间:2023-12-04 20:35:46

快速排序算法是一种常用的排序算法。它的基本思想是选择一个基准元素,将待排序序列分成两部分,一部分小于等于基准元素,一部分大于等于基准元素,然后对这两部分序列分别进行排序,最后将两部分序列合并起来。

下面是使用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]  # 将小于等于基准元素的元素放入less列表
        greater = [x for x in arr[1:] if x > pivot]  # 将大于基准元素的元素放入greater列表
        return quick_sort(less) + [pivot] + quick_sort(greater)  # 递归调用快速排序并合并结果

# 使用例子
if __name__ == "__main__":
    arr = [5, 3, 8, 4, 2, 9, 1, 7, 6]
    sorted_arr = quick_sort(arr)
    print(sorted_arr)

上述代码中,quick_sort函数实现了快速排序算法。它首先判断输入的列表arr的长度是否小于等于1,如果是,则直接返回该列表;否则,选择第一个元素作为基准元素,然后以基准元素为准将序列分成两部分,将小于等于基准元素的元素放入less列表,将大于基准元素的元素放入greater列表。最后,使用递归调用快速排序并合并结果,最终返回排序后的列表。

在使用例子中,我们声明一个未排序的列表arr,然后调用quick_sort函数对其进行排序,并将排序结果打印出来。输出结果为[1, 2, 3, 4, 5, 6, 7, 8, 9],表示排序成功。

快速排序算法的时间复杂度平均为O(nlogn),空间复杂度为O(logn)。因此,它在大多数情况下具有较高的效率。