add bead sort algorithm

dori-dev · dori-dev · commit 685aa2e22b33 · 2022-08-03T10:55:44.000+04:30
diff --git a/arrays/bead_sort.py b/arrays/bead_sort.py
@@ -0,0 +1,25 @@
+"""
+array = [4, 7, 9, 2, 3, 6, 7]
+bead_sort(array) => [2, 3, 4, 6, 7, 7, 9]
+"""
+
+from typing import List
+
+
+def bead_sort(sequence: List[int]) -> List[int]:
+    for number in sequence:
+        if not isinstance(number, int) or number < 0:
+            raise TypeError('Sequence must be list or non-negative integers.')
+    for _ in range(len(sequence)):
+        for i, (first, second) in enumerate(zip(sequence, sequence[1:])):
+            if second < first:
+                distance = first - second
+                sequence[i] -= distance
+                sequence[i+1] += distance
+    return sequence
+
+
+if __name__ == "__main__":
+    array = [4, 7, 9, 2, 3, 6, 7]
+    bead_sort(array)
+    print(array == [2, 3, 4, 6, 7, 7, 9])
