Rotate Array

update Sep 9,2017 23:00

LeetCode

Rotate an array of n elements to the right by k steps.

For example, with n = 7 and k = 3, the array [1,2,3,4,5,6,7] is rotated to [5,6,7,1,2,3,4].

Note: Try to come up as many solutions as you can, there are at least 3 different ways to solve this problem.

Hint: Could you do it in-place with O(1) extra space?

Basic Idea:

方法1： 最简单的思路就是把nums复制一遍，然后分两段复制回去。

Java Code:

class Solution {
    public void rotate(int[] nums, int k) {
        k %= nums.length; 
        int[] arr = Arrays.copyOf(nums, nums.length);
        System.arraycopy(arr, 0, nums, k, nums.length - k);
        System.arraycopy(arr, nums.length - k, nums, 0, k);
    }
}

方法2： 例如对于 [1,2,3,4,5,6,7], k=3, rotate 之后为 [5,6,7,1,2,3,4]. 仔细观察上面的两个数组，我们就会发现一个规律，将结果reverse之后是 [4,3,2,1,7,6,5]，其实就是对于长度为n的数组，先将前 n-k 个reverse，然后将后 k 个reverse，然后整体 reverse，就得到了最终结果。

Java Code:

    class Solution {
        public void rotate(int[] nums, int k) {
            int n = nums.length;
            k = k % n;
            reverse(nums, 0, n - k - 1); // reverse first part
            reverse(nums, n - k, n - 1); // reverse second part
            reverse(nums, 0, n - 1);     // reverse the whole array
        }
        private void reverse(int[] nums, int i, int j) {
            if (i < 0 || i >= j) return;
            while (i < j) {
                int t = nums[i];
                nums[i] = nums[j];
                nums[j] = t;
                i++;
                j--;
            }
        }
    }

Python Code:

    class Solution:
        def rotate(self, nums, k):
            """
            :type nums: List[int]
            :type k: int
            :rtype: void Do not return anything, modify nums in-place instead.
            """
            def rotate(left, right):
                if left >= right:
                    return
                while left < right:
                    t = nums[left]
                    nums[left] = nums[right]
                    nums[right] = t
                    left += 1
                    right -= 1

            k = k % len(nums)
            rotate(0, len(nums) - k - 1)
            rotate(len(nums) - k, len(nums) - 1)
            rotate(0, len(nums) - 1)

update May 6,2018 2:37

C++ Code

C++ 中 stl 内置了 reverse 函数，就不需要再自己写reverse函数了。

    class Solution {
    public:
        void rotate(vector<int>& nums, int k) {
            k %= nums.size();
            reverse(nums.begin(), nums.begin() + nums.size() - k);
            reverse(nums.begin() + nums.size() - k, nums.end());
            reverse(nums.begin(), nums.end());
        }
    };