def sort1(arr): """ 思路: 以arr[0]为pivot 以arr长度小于等于1为边界,返回arr 分别将小于pivot、等于pivot、大于pivot的分类 递归处理两边的分类,将结果组合返回 :param arr: :return: """ if len(arr)
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) 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))