快速排序的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))
本文标题:快速排序的python实现
地址分享:http://hbruida.cn/article/pcjcii.html