Skip to content

Commit

Permalink
Adding quick sort where random pivot point is chosen (TheAlgorithms#774)
Browse files Browse the repository at this point in the history
  • Loading branch information
gattlin1 authored and poyea committed Apr 30, 2019
1 parent 48553da commit 06dbef0
Showing 1 changed file with 33 additions and 0 deletions.
33 changes: 33 additions & 0 deletions sorts/random_pivot_quick_sort.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
"""
Picks the random index as the pivot
"""
import random

def partition(A, left_index, right_index):
pivot = A[left_index]
i = left_index + 1
for j in range(left_index + 1, right_index):
if A[j] < pivot:
A[j], A[i] = A[i], A[j]
i += 1
A[left_index], A[i - 1] = A[i - 1], A[left_index]
return i - 1

def quick_sort_random(A, left, right):
if left < right:
pivot = random.randint(left, right - 1)
A[pivot], A[left] = A[left], A[pivot] #switches the pivot with the left most bound
pivot_index = partition(A, left, right)
quick_sort_random(A, left, pivot_index) #recursive quicksort to the left of the pivot point
quick_sort_random(A, pivot_index + 1, right) #recursive quicksort to the right of the pivot point

def main():
user_input = input('Enter numbers separated by a comma:\n').strip()
arr = [int(item) for item in user_input.split(',')]

quick_sort_random(arr, 0, len(arr))

print(arr)

if __name__ == "__main__":
main()

0 comments on commit 06dbef0

Please sign in to comment.