diff --git a/data_structures/arrays/rotate_array.py b/data_structures/arrays/rotate_array.py index 6d5033fb7..8635f35a1 100644 --- a/data_structures/arrays/rotate_array.py +++ b/data_structures/arrays/rotate_array.py @@ -58,7 +58,8 @@ def rotate_array_n_space(nums: list[int], k: int) -> list[int]: >>> rotate_array_n_space([8, -7, 22, -6], 9) [-6, 8, -7, 22] """ - + length = len(nums) + k = k % length return nums[-k:] + nums[:k]