Вот более четкая реализация шага разделения:
def quicksort(arr, low, high):
if low > high or low == high:
return
pivot = randint(low, high)
updated_pivot = partition(pivot,arr, low, high)
quicksort(arr, low, updated_pivot-1)
quicksort(arr, updated_pivot+1, high)
def partition(pivot, arr, low, high):
arr[low], arr[pivot] = arr[pivot], arr[low] #now pivot is at 'low' index of current subarray
start_of_ = 1
curr = 1
while curr <= high:
if arr[curr] <= arr[low]:
arr[start], arr[curr] = arr[curr], arr[start] #making sure all values between index low and index start (exclusive) are less than or equal to the pivot.
start+=1
curr += 1
new_pivot_location = start - 1 #the value at index start is the first value greater than the pivot (the range considered in the while loop is exclusive)
arr[new_pivot_location], arr[low] =arr[low], arr[new_pivot_location]
return new_pivot_location
ПРИМЕР ЗАПУСКА:
Ввод:
[5,1,3,8, 0,2]
|
pivot
Алгоритм разделения:
[3,1,5,8,0,2] --> after switching pivot's position
|
pivot
start
|
[3,1,5,8,0,2] --> initializing start and curr
|
curr
start
|
[3,1,5,8,0,2] --> 1 < 3, so we swap 1 & 1, and start++, curr++
|
curr
start
|
[3,1,5,8,0,2] --> 5 > 3, so we don't swap. Don't move start, curr++
|
curr
start
|
[3,1,5,8,0,2] --> 8 > 3, so we don't swap. Don't move start, curr++
|
curr
start
|
[3,1,0,8,5,2] --> 0 < 3, so we swap 5 and 0. start++, curr++
|
curr
start
|
[3,1,0,2,5,8] --> 2 < 3, so we swap 8 and 2. start++, curr++
|
curr
[2,1,0,3,5,8] --> swap 2 and 3, and reset pivot index.
Выход:
[2,1,0,3,5,8]
|
pivot