python实现快速排序

算法导论中伪代码

   快速排序的思想是任找一个元素作为基准(pivot),对待排数组进行分组,使基准元素左边的数据比基准数据要小,右边的数据比基准数据要大,这样基准元素就放在了正确的位置上。然后对基准元素左边和右边的组进行相同的操作,最后将数据排序完成。

   这里partition相当于用快慢指针实现了把数组A分成两部分,A[left..i] ≤ pivot,A[i+1..j] > pivot。还有很多其他实现方法,比如一个指针从前往后,一个指针从后往前,每次判断是否交换并移动指针。

def partition(arr, left, right):
    pivot = arr[right]
    i = left - 1
    for j in range(left, right):
        if arr[j] <= pivot:
            i += 1
            arr[i], arr[j] = arr[j], arr[i]
    arr[i+1], arr[right] = arr[right], arr[i+1]
    return i + 1

def quick_sort(arr, left, right):
    if left < right:
        q = partition(arr, left, right)
        quick_sort(arr, left, q - 1)
        quick_sort(arr, q + 1, right)

arr = [-2, 1, -3, 4, -1, 2, 1, -5, 4]
quick_sort(arr, 0, len(arr)-1)
print(arr)
posted @ 2020-05-04 00:26  肃木易  阅读(374)  评论(0编辑  收藏  举报