Skip to content
Closed
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
26 changes: 22 additions & 4 deletions sorts/shell_sort.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,28 +2,46 @@
https://en.wikipedia.org/wiki/Shellsort#Pseudocode
"""

from typing import Any, Protocol, TypeVar


class Comparable(Protocol):
def __lt__(self, other: Any, /) -> bool: ...


T = TypeVar("T", bound=Comparable)


def shell_sort(collection: list[T]) -> list[T]:

Check failure on line 15 in sorts/shell_sort.py

View workflow job for this annotation

GitHub Actions / ruff

ruff (UP047)

sorts/shell_sort.py:15:5: UP047 Generic function `shell_sort` should use type parameters help: Use type parameters

Check failure on line 15 in sorts/shell_sort.py

View workflow job for this annotation

GitHub Actions / ruff

ruff (UP047)

sorts/shell_sort.py:15:5: UP047 Generic function `shell_sort` should use type parameters help: Use type parameters
"""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:
Expand Down
Loading