数据结构与算法 Python实现 快速排序

快速排序分为三步

• 选择基准值，pirot将数组分为两个字数组，小于基准值的元素和大语句准直的元素，这个过程称之为prtition
• 对这两个数组进行排序
• 合并结果

def quicksort(array):
    #当排序的数组小于2就不排序
    if len(array) < 2:
        return array
    else:
        #定一下标
        pivot_index=0
        #取到值
        pivot = array[pivot_index]
        #小于他的值给less_part
        less_part = [i for i in array[pivot_index+1 :] if i <= pivot]
        #大于的值给great_part
        great_part = [i for i in array[pivot_index+1 :] if i > pivot]
        #注意这里的递归参数传的是小于分割点的集合
        return quicksort(less_part)+[pivot]+quicksort(great_part)

def test_quicksort():
    #生成随机数
    import  random
    #创建一个新的list
    seq = list(range(10))
    #随机取出seq
    random.shuffle(seq)
    #assert 测试一个线程 ，sorted返回一个新的list
    assert quicksort(seq) == sorted(seq)
#找到最合适分割值
def partition(array,beg ,end):
    pivot_index = beg
    pivot = array[pivot_index]
    left = pivot_index + 1
    right = end - 1

    while True:
        while left <= right and array[left] < pivot:
            left += 1
        while right >= left and array[right] >= pivot:
            right -= 1
        if left > right:
            break
        else:
            array[left],array[right] = array[right],array[left]
    array[pivot_index],array[right] = array[right],array[pivot_index]
    return right