Create radix_sort.py

added python implementation of radix sort

src/python/radix_sort.py

@@ -0,0 +1,43 @@
+def radix_sort(arr):
+ # Find the maximum number to know the number of digits
+ max_num = max(arr)
+
+ # Do counting sort for every digit, starting from the least significant digit
+ exp = 1
+ while max_num // exp > 0:
+ counting_sort(arr, exp)
+ exp *= 10
+
+
+def counting_sort(arr, exp):
+ n = len(arr)
+ output = [0] * n
+ count = [0] * 10
+
+ for i in range(n):
+ index = arr[i] // exp
+ count[index % 10] += 1
+
+ for i in range(1, 10):
+ count[i] += count[i - 1]
+
+ i = n - 1
+ while i >= 0:
+ index = arr[i] // exp
+ output[count[index % 10] - 1] = arr[i]
+ count[index % 10] -= 1
+ i -= 1
+
+ for i in range(n):
+ arr[i] = output[i]
+
+
+def main():
+ arr = [170, 2, 45, 75, 75, 90, 802, 24, 2, 66]
+ print("Unsorted array:", arr)
+ radix_sort(arr)
+ print("Sorted array:", arr)
+
+
+if __name__ == "__main__":
+ main()