-
Notifications
You must be signed in to change notification settings - Fork 8
/
Copy pathmain.py
40 lines (28 loc) · 768 Bytes
/
main.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
import random
def radix_sort(arr):
RADIX, digit = 10, 1
max_digit = max(arr)
while digit < max_digit:
buckets = [list() for _ in range( RADIX )]
for num in arr:
n = (num // digit) % RADIX
buckets[n].append(num)
idx = 0
for i in range( RADIX ):
bucket = buckets[i]
for val in bucket:
arr[idx] = val
idx += 1
digit *= 10
def main():
num = 20
range_left = 0
range_right = 10000
arr = [random.randint(range_left, range_right) for _ in range(num)]
print('Original array:')
print(arr)
radix_sort(arr)
print('Sorted array:')
print(arr)
if __name__ == '__main__':
main()