add rotate algorithm

dori-dev · dori-dev · commit 1f2bb52a6532 · 2022-08-03T16:11:00.000+04:30
diff --git a/strings/rotate.py b/strings/rotate.py
@@ -0,0 +1,21 @@
+"""
+rotate("hello", 2) => "llohe"
+rotate("hello", 5) => "hello"
+rotate("hello", 6) => "elloh"
+rotate("hello", 7) => "llohe"
+complexity: O(1)
+"""
+
+
+def rotate(string: str, degree: int) -> str:
+    degree %= len(string)
+    double_string = string + string
+    return double_string[degree:degree+len(string)]
+
+
+if __name__ == "__main__":
+    print(rotate("hello", 2) == "llohe")
+    print(rotate("hello", 5) == "hello")
+    print(rotate("hello", 6) == "elloh")
+    print(rotate("hello", 7) == "llohe")
+    print(rotate("hello", 102) == "llohe")
