- Notifications
You must be signed in to change notification settings - Fork 46.7k
/
Copy pathshell_sort.py
40 lines (33 loc) · 1.13 KB
/
shell_sort.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
"""
https://en.wikipedia.org/wiki/Shellsort#Pseudocode
"""
defshell_sort(collection: list[int]) ->list[int]:
"""Pure implementation of shell sort algorithm in Python
:param collection: Some mutable ordered collection with heterogeneous
comparable items inside
:return: the same collection ordered by ascending
>>> shell_sort([0, 5, 3, 2, 2])
[0, 2, 2, 3, 5]
>>> shell_sort([])
[]
>>> shell_sort([-2, -5, -45])
[-45, -5, -2]
"""
# Marcin Ciura's gap sequence
gaps= [701, 301, 132, 57, 23, 10, 4, 1]
forgapingaps:
foriinrange(gap, len(collection)):
insert_value=collection[i]
j=i
whilej>=gapandcollection[j-gap] >insert_value:
collection[j] =collection[j-gap]
j-=gap
ifj!=i:
collection[j] =insert_value
returncollection
if__name__=="__main__":
fromdoctestimporttestmod
testmod()
user_input=input("Enter numbers separated by a comma:\n").strip()
unsorted= [int(item) foriteminuser_input.split(",")]
print(shell_sort(unsorted))