keon · goswami-rahul · Jun 6, 2018 · Jun 5, 2018 · Jun 5, 2018 · Jun 5, 2018
diff --git a/README.md b/README.md
@@ -222,6 +222,7 @@ If you want to uninstall algorithms, it is as simple as:
 - [sort](algorithms/sort)
     - [bubble_sort](algorithms/sort/bubble_sort.py)
     - [bucket_sort](algorithms/sort/bucket_sort.py)
+    - [cocktail_shaker_sort](algorithms/sort/cocktail_shaker_sort.py)
     - [comb_sort](algorithms/sort/comb_sort.py)
     - [counting_sort](algorithms/sort/counting_sort.py)
     - [heap_sort](algorithms/sort/heap_sort.py)

diff --git a/README_CN.md b/README_CN.md
@@ -204,6 +204,7 @@ pip3 uninstall -y algorithms
 - [sort：排序](algorithms/sort)
     - [bubble_sort：冒泡排序](algorithms/sort/bubble_sort.py)
     - [bucket_sort](algorithms/sort/bucket_sort.py)
+    - [cocktail_shaker_sort](algorithms/sort/cocktail_shaker_sort.py)
     - [comb_sort：梳排序](algorithms/sort/comb_sort.py)
     - [counting_sort：计数排序](algorithms/sort/counting_sort.py)
     - [heap_sort：堆排序](algorithms/sort/heap_sort.py)

diff --git a/README_GE.md b/README_GE.md
@@ -229,6 +229,7 @@ Um das Projekt zu deinstallieren tippen Sie folgendes:
 - [sort](algorithms/sort)
     - [bubble_sort](algorithms/sort/bubble_sort.py)
     - [bucket_sort](algorithms/sort/bucket_sort.py)
+    - [cocktail_shaker_sort](algorithms/sort/cocktail_shaker_sort.py)
     - [comb_sort](algorithms/sort/comb_sort.py)
     - [counting_sort](algorithms/sort/counting_sort.py)
     - [heap_sort](algorithms/sort/heap_sort.py)

diff --git a/README_JP.md b/README_JP.md
@@ -219,6 +219,7 @@ if __name__ == "__main__":
 - [sort : ソート](algorithms/sort)
     - [bubble_sort](algorithms/sort/bubble_sort.py)
     - [bucket_sort](algorithms/sort/bucket_sort.py)
+    - [cocktail_shaker_sort](algorithms/sort/cocktail_shaker_sort.py)
     - [comb_sort](algorithms/sort/comb_sort.py)
     - [counting_sort](algorithms/sort/counting_sort.py)
     - [heap_sort](algorithms/sort/heap_sort.py)

diff --git a/README_KR.md b/README_KR.md
@@ -214,6 +214,7 @@ if __name__ == "__main__":
     - [set_covering](algorithms/set/set_covering.py)
 - [sort : 정렬 알고리즘](algorithms/sort)
     - [bubble_sort](algorithms/sort/bubble_sort.py)
+    - [cocktail_shaker_sort](algorithms/sort/cocktail_shaker_sort.py)
     - [comb_sort](algorithms/sort/comb_sort.py)
     - [counting_sort](algorithms/sort/counting_sort.py)
     - [heap_sort](algorithms/sort/heap_sort.py)

diff --git a/algorithms/sort/__init__.py b/algorithms/sort/__init__.py
@@ -10,3 +10,4 @@
 from .bucket_sort import *
 from .shell_sort import *
 from .radix_sort import *
+from .cocktail_shaker_sort import *
diff --git a/algorithms/sort/cocktail_shaker_sort.py b/algorithms/sort/cocktail_shaker_sort.py
@@ -0,0 +1,30 @@
+def cocktail_shaker_sort(arr):
+    """
+    Cocktail_shaker_sort
+    Sorting a given array
+    mutation of bubble sort
+
+    reference: https://en.wikipedia.org/wiki/Cocktail_shaker_sort
+
+    Worst-case performance: O(N^2)
+    """
+
+    def swap(i, j):
+        arr[i], arr[j] = arr[j], arr[i]
+
+    n = len(arr)
+    swapped = True
+    while swapped:
+        swapped = False
+        for i in range(1, n):
+            if arr[i - 1] > arr[i]:
+                swap(i - 1, i)
+                swapped = True
+        if swapped == False:
+            return arr
+        swapped = False
+        for i in range(n-1,0,-1):
+            if arr[i - 1] > arr[i]:
+                swap(i - 1, i)
+                swapped = True
+    return arr
diff --git a/tests/test_sort.py b/tests/test_sort.py
@@ -9,7 +9,8 @@
     selection_sort,
     bucket_sort,
     shell_sort,
-    radix_sort
+    radix_sort,
+    cocktail_shaker_sort
 )
 
 import unittest
@@ -63,7 +64,10 @@ def test_shell_sort(self):
     def test_radix_sort(self):
         self.assertEqual([1, 5, 23, 57, 65, 1232],
                         radix_sort([1, 5, 65, 23, 57, 1232]))
-
+
+    def test_cocktail_shaker_sort(self):
+        self.assertEqual([1, 5, 23, 57, 65, 1232],
+                        cocktail_shaker_sort([1, 5, 65, 23, 57, 1232]))
 
 
 if __name__ == "__main__":