快速排序的python实现

def sort1(arr):
    """
    思路:
    以arr[0]为pivot
    以arr长度小于等于1为边界，返回arr
    分别将小于pivot、等于pivot、大于pivot的分类
    递归处理两边的分类，将结果组合返回
    :param arr:
    :return:
    """
    if len(arr) <= 1:
        return arr
    pivot = arr[0]
    left = [x for x in arr if x < pivot]
    middle = [x for x in arr if x == pivot]
    right = [x for x in arr if x > pivot]
    return sort1(right) + middle + sort1(left)


def sort2(arr, arr_l, arr_r):
    """
    思路：
    以arr[arr_r]为pivot
    以arr长度小于等于1为边界，直接返回
    左游标从arr_l到arr_r移动，当arr[左游标]<=pivot时进行处理：
        if arr[左游标]<=arr[r]:
            if 左游标 == arr_r:
                递归处理 arr_l到arr_r-1
            else:
                右游标从arr_r-1到左游标移动：
                    if 右游标>左游标 and arr[右游标]>pivot:
                        交换arr[左游标] arr[右游标]
                        跳出右游标的循环
                    elif 右游标 == 左游标:
                        交换arr[右游标] pivot
                        递归处理 arr_l到(右游标-1)
                        递归处理 (右游标+1)到arr_r


    :param arr:
    :param arr_l:
    :param arr_r:
    :return:
    """
    if len(arr) <= 1:
        return
    for left in range(arr_l, arr_r+1):
        if arr[left] <= arr[arr_r]:
            if left == arr_r:
                sort2(arr, arr_l, arr_r-1)
            else:
                for right in range(arr_r-1, left-1, -1):
                    if right > left and arr[right] > arr[arr_r]:
                        arr[right], arr[left] = arr[left], arr[right]
                        break
                    elif right == left:
                        arr[right], arr[arr_r] = arr[arr_r], arr[right]
                        sort2(arr, arr_l, right-1)
                        sort2(arr, right+1, arr_r)
                        return


def sort(arr, method=2):
    if method == 1:
        return sort1(arr)
    elif method == 2:
        sort2(arr, 0, len(arr)-1)
        return arr


if __name__ == "__main__":
    l = [5, 2, 7, 8, 6, 1, 4, 9, 10, 1, 2, 3, 4]
    print(sort(l))