快速排序的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))