TheAlgorithms-Python
48 строк · 1.2 Кб
1"""
2This is a pure Python implementation of the radix sort algorithm
3
4Source: https://en.wikipedia.org/wiki/Radix_sort
5"""
6
7from __future__ import annotations
8
9RADIX = 10
10
11
12def radix_sort(list_of_ints: list[int]) -> list[int]:
13"""
14Examples:
15>>> radix_sort([0, 5, 3, 2, 2])
16[0, 2, 2, 3, 5]
17
18>>> radix_sort(list(range(15))) == sorted(range(15))
19True
20>>> radix_sort(list(range(14,-1,-1))) == sorted(range(15))
21True
22>>> radix_sort([1,100,10,1000]) == sorted([1,100,10,1000])
23True
24"""
25placement = 1
26max_digit = max(list_of_ints)
27while placement <= max_digit:
28# declare and initialize empty buckets
29buckets: list[list] = [[] for _ in range(RADIX)]
30# split list_of_ints between the buckets
31for i in list_of_ints:
32tmp = int((i / placement) % RADIX)
33buckets[tmp].append(i)
34# put each buckets' contents into list_of_ints
35a = 0
36for b in range(RADIX):
37for i in buckets[b]:
38list_of_ints[a] = i
39a += 1
40# move to next
41placement *= RADIX
42return list_of_ints
43
44
45if __name__ == "__main__":
46import doctest
47
48doctest.testmod()
49