diff --git a/sorts/shell_sort.py b/sorts/shell_sort.py index b65609c974b7..e3805c5ac6f6 100644 --- a/sorts/shell_sort.py +++ b/sorts/shell_sort.py @@ -1,13 +1,15 @@ -""" +"""Shell sort implementation. + +Reference: https://en.wikipedia.org/wiki/Shellsort#Pseudocode """ def shell_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 + """Sort a list of integers in ascending order using shell sort. + + :param collection: A list of integers + :return: The same list sorted in ascending order >>> shell_sort([0, 5, 3, 2, 2]) [0, 2, 2, 3, 5] @@ -16,18 +18,19 @@ def shell_sort(collection: list[int]) -> list[int]: >>> shell_sort([-2, -5, -45]) [-45, -5, -2] """ - # Marcin Ciura's gap sequence - gaps = [701, 301, 132, 57, 23, 10, 4, 1] + for gap in gaps: for i in range(gap, len(collection)): insert_value = collection[i] j = i + while j >= gap and collection[j - gap] > insert_value: collection[j] = collection[j - gap] j -= gap - if j != i: - collection[j] = insert_value + + collection[j] = insert_value + return collection