用Python实现快速排序

寻技术 Python编程 2023年08月09日 87

快速排序是一种常用的排序算法,其时间复杂度为 O(nlogn)。在实际应用中,快速排序通常比其他排序算法快得多。Python 提供了许多内置的排序函数,但了解和实现快速排序仍然很重要。在本文中,我们将通过 Python 实现快速排序算法。

快速排序的工作原理是选定一个基准值(pivot),然后将列表中所有小于基准值的元素放在一个子列表中,将所有大于基准值的元素放在另一个子列表中。然后对这两个子列表递归进行快速排序。最终,所有子列表都将被递归排序,然后合并成一个排好序的列表。

以下是用 Python 实现快速排序的代码:

def quick_sort(arr):
    if len(arr) < 2:
        return arr
    else:
        pivot = arr[0]
        less = [i for i in arr[1:] if i <= pivot]
        greater = [i for i in arr[1:] if i > pivot]
        return quick_sort(less) + [pivot] + quick_sort(greater)

在上面的代码中,我们首先检查列表的长度。如果列表长度小于 2,我们就返回原列表。否则,我们选择列表的第一个元素作为基准值(pivot)。然后,我们使用列表推导式将小于等于基准值的元素放入一个列表中,并将大于基准值的元素放入另一个列表中。接下来,我们递归将较小和较大的列表进行排序,并将排好序的列表连接在一起,基准值置于连接的列表中间。

这个算法需要选择一个合适的基准数。如果选择的基准数恰好是列表中的最大(或最小)值,那么递归排序的子列表大小只减少了 1。这种情况下,快速排序算法的时间复杂度可能会退化为 O(n2)。为了避免这种情况,我们可以使用随机选取基准值的方法。这个方法在基准值不是列表中的最大(或最小)值的情况下,平均地保证了递归排序的子列表大小。

以下是使用随机选择基准值的 Python 代码:

import random

def quick_sort(arr):
    if len(arr) < 2:
        return arr
    else:
        pivot_index = random.randint(0, len(arr) - 1)
        pivot = arr[pivot_index]
        less = [i for i in arr[:pivot_index] + arr[pivot_index + 1:] if i <= pivot]
        greater = [i for i in arr[:pivot_index] + arr[pivot_index + 1:] if i > pivot]
        return quick_sort(less) + [pivot] + quick_sort(greater)

在上面的代码中,我们先使用 random.randint() 函数随机选择一个基准值。然后,我们将小于等于基准值的元素放入一个列表中,并将大于基准值的元素放入另一个列表中,这与前面那个实现方法是类似的。

总结一下,我们通过 Python 实现了快速排序算法,并使用随机选择基准值的方法避免了递归排序的子列表的大小不均衡的情况。这个算法是基于分治(Divide and Conquer)思想的,它能够在平均情况下以 O(nlogn) 的时间复杂度快速对列表进行排序。

关闭

用微信“扫一扫”