您的位置:

Python实现快速排序算法

一、什么是快速排序算法

快速排序算法(Quick Sort),是一种高效的、比较常用的排序算法。快速排序主要思想是通过选择一个"基准"数,将待排序序列分成两个子序列,在这两个子序列中分别对每个元素进行比较和排序,从而达到排序整个序列的目的。

二、快速排序算法的原理

快速排序的原理很简单,其主要分为三步:

1. 选择一个数作为基准数,一般为第一个数。

2. 分区过程:将原序列分为两个子序列,小于基准数的放在左边,大于等于基准数的放在右边。分区的过程可以使用快慢指针或者左右指针来实现。

3. 递归实现:递归地对左右两个子序列进行同样的操作,直到所有子序列的长度都为 1。

三、代码实现

def quick_sort(arr, left, right):
    if left >= right:
        return
    i = left
    j = right
    base = arr[i] # 选择第一个数为基准数
    while i < j:
        while i < j and arr[j] >= base:
            j -= 1
        arr[i] = arr[j]
        while i < j and arr[i] < base:
            i += 1
        arr[j] = arr[i]
    arr[i] = base
    quick_sort(arr, left, i - 1)
    quick_sort(arr, i + 1, right)

四、时间复杂度分析

在最优情况下,每次划分所得的两个子序列的长度大致相等,此时快排的时间复杂度为 O(nlogn),其中n为数组的长度。

在最坏情况下,每次划分所得的两个子序列的长度分别为 1 和 (n-1),此时快排的时间复杂度为 O(n^2)。因此,为了避免最坏情况的出现,需要采取一些优化措施。

五、快速排序的优化

1. 随机选择基准数,避免最坏情况的出现。

import random

def quick_sort(arr, left, right):
    if left >= right:
        return
    i = left
    j = right
    base = arr[random.randint(left, right)]
    while i < j:
        while i < j and arr[j] >= base:
            j -= 1
        arr[i] = arr[j]
        while i < j and arr[i] < base:
            i += 1
        arr[j] = arr[i]
    arr[i] = base
    quick_sort(arr, left, i - 1)
    quick_sort(arr, i + 1, right)

2. 当序列长度小于某一个值时,采用插入排序。

def insert_sort(arr, left, right):
    for i in range(left+1, right+1):
        j = i
        while j > left and arr[j] < arr[j-1]:
            arr[j], arr[j-1] = arr[j-1], arr[j]
            j -= 1

def quick_sort(arr, left, right):
    if left >= right:
        return
    if right - left <= 10:
        insert_sort(arr, left, right)
        return
    i = left
    j = right
    base = arr[random.randint(left, right)]
    while i < j:
        while i < j and arr[j] >= base:
            j -= 1
        arr[i] = arr[j]
        while i < j and arr[i] < base:
            i += 1
        arr[j] = arr[i]
    arr[i] = base
    quick_sort(arr, left, i - 1)
    quick_sort(arr, i + 1, right)

六、总结

快速排序是一种非常高效的排序算法,其原理简单,代码实现也比较容易。但是,在最坏情况下,快排的时间复杂度会达到 O(n^2),因此需要采取一些优化措施来避免最坏情况的出现。

通过对快速排序算法的优化,可以提高算法的效率。在实际应用中,快排算法经常被使用,如对数据进行排序、在数据处理场景中查找中位数等。