mirror of
https://github.com/TheAlgorithms/Python.git
synced 2026-03-13 09:50:19 +08:00
Sortings and searching algorithms refactoring
This commit is contained in:
61
quick_sort.py
Normal file
61
quick_sort.py
Normal file
@@ -0,0 +1,61 @@
|
||||
"""
|
||||
This is pure python implementation of quick sort algorithm
|
||||
|
||||
For doctests run following command:
|
||||
python -m doctest -v quick_sort.py
|
||||
or
|
||||
python3 -m doctest -v quick_sort.py
|
||||
|
||||
For manual testing run:
|
||||
python quick_sort.py
|
||||
"""
|
||||
from __future__ import print_function
|
||||
|
||||
|
||||
def quick_sort(collection):
|
||||
"""Pure implementation of quick sort algorithm in Python
|
||||
|
||||
:param collection: some mutable ordered collection with heterogeneous
|
||||
comparable items inside
|
||||
:return: the same collection ordered by ascending
|
||||
|
||||
Examples:
|
||||
>>> quick_sort([0, 5, 3, 2, 2])
|
||||
[0, 2, 2, 3, 5]
|
||||
|
||||
>>> quick_sort([])
|
||||
[]
|
||||
|
||||
>>> quick_sort([-2, -5, -45])
|
||||
[-45, -5, -2]
|
||||
"""
|
||||
less = []
|
||||
equal = []
|
||||
greater = []
|
||||
if len(collection) > 1:
|
||||
pivot = collection[0]
|
||||
for x in collection:
|
||||
if x < pivot:
|
||||
less.append(x)
|
||||
if x == pivot:
|
||||
equal.append(x)
|
||||
if x > pivot:
|
||||
greater.append(x)
|
||||
return quick_sort(less) + equal + quick_sort(greater)
|
||||
else:
|
||||
return collection
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
import sys
|
||||
|
||||
# For python 2.x and 3.x compatibility: 3.x has not raw_input builtin
|
||||
# otherwise 2.x's input builtin function is too "smart"
|
||||
if sys.version_info.major < 3:
|
||||
input_function = raw_input
|
||||
else:
|
||||
input_function = input
|
||||
|
||||
user_input = input_function('Enter numbers separated by coma:\n')
|
||||
unsorted = [int(item) for item in user_input.split(',')]
|
||||
print(quick_sort(unsorted))
|
||||
Reference in New Issue
Block a user