fix_rotate.py (keon#778)

TrentonSouth · web-flow · commit 339a0d276a2b · 2021-05-12T14:03:55.000-07:00
* fix_rotate.py

The way it was, you couldn't pass in a rotation higher than the length of the string doubled without getting a blank string in return. This way allows you to pass in any positive integer and get a result.

Also,
The last line of the description said that it took 2 strings and returned a boolean. This is not correct. The first line of the comment is correct.

* Update rotate.py

After submitting my last change, I realized it would have been better to not keep concatenating the string.

* Updated Comments rotate.py
diff --git a/algorithms/strings/rotate.py b/algorithms/strings/rotate.py
@@ -1,18 +1,19 @@
 """
 Given a strings s and int k, return a string that rotates k times
 
+k can be any positive integer.
+
 For example,
 rotate("hello", 2) return "llohe"
 rotate("hello", 5) return "hello"
 rotate("hello", 6) return "elloh"
 rotate("hello", 7) return "llohe"
+rotate("hello", 102) return "lohel"
 
-accepts two strings
-returns bool
 """
 def rotate(s, k):
-    double_s = s + s
+    long_string = s * (k // len(s) + 2)
     if k <= len(s):
-        return double_s[k:k + len(s)]
+        return long_string[k:k + len(s)]
     else:
-        return double_s[k-len(s):k]
+        return long_string[k-len(s):k]
