diff --git a/sorts/shell_sort.py b/sorts/shell_sort.py index b65609c974b7..6658112af9dc 100644 --- a/sorts/shell_sort.py +++ b/sorts/shell_sort.py @@ -2,28 +2,43 @@ https://en.wikipedia.org/wiki/Shellsort#Pseudocode """ +from typing import Any, Protocol + + +class Comparable(Protocol): + def __lt__(self, other: Any, /) -> bool: ... + + +def shell_sort[T: Comparable](collection: list[T]) -> list[T]: + """Pure implementation of shell sort algorithm in Python. -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 + Examples: >>> shell_sort([0, 5, 3, 2, 2]) [0, 2, 2, 3, 5] >>> shell_sort([]) [] >>> shell_sort([-2, -5, -45]) [-45, -5, -2] + >>> shell_sort(["c", "a", "b"]) + ['a', 'b', 'c'] + >>> shell_sort([2.5, -1.0, 0.0]) + [-1.0, 0.0, 2.5] + >>> shell_sort([0, 5, 3, 2, 2]) == sorted([0, 5, 3, 2, 2]) + True + >>> shell_sort(["c", "a", "b"]) == sorted(["c", "a", "b"]) + True """ # 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: + while j >= gap and insert_value < collection[j - gap]: collection[j] = collection[j - gap] j -= gap if j != i: