[LeetCode]189. Rotate Array

189. Rotate Array

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].

分析

题目需要旋转一个数组，使得右边的k个元素不变序的搬到数组头部。
方法一：将后面k个元素逐个往前移动n-k步，这种方式不适用额外的空间，但是提交后会超时。
方法二：适用STL的insert方法，逐个将后k个元素插入到头部，并将该元素从尾部删除
方法三：将整个序列翻转，然后分别翻转前面部分和后面部分

源码

方法一：

    void rotate(vector<int>& nums, int k) {
        // 将k之后的数和它前面的数逐个交换
        int length = nums.size();
        k = k % length;
        int steps = length - k; // k和前面交换steps次就可以了
        for(int i = (length - k); i < length; i++) {
            int stop = i - steps; // 交换的截止位置
            int tmp = nums[i];
            int j = i;
            while(j > stop) {
                nums[j] = nums[j-1]; // 往后移动数字
                j--;
            }
            nums[j] = tmp; // 将K数放到前面
        }
    }

方法二：

    void rotate(vector<int>& nums, int k) {
        for (int i = 0; i < k; ++i) {
            nums.insert(nums.begin(), nums[nums.size() - 1]);
            nums.pop_back();
        }
    }

方法三：

    void rotate(vector<int>& nums, int k) {
        int length = nums.size();
        k = k % length; // 会出现k > length的情况
        if(k == 0 || k >= length) return;
        jk_reverse(nums,0,length);
        jk_reverse(nums,0,k);
        jk_reverse(nums,k,length - k);
    }
    void jk_reverse(vector<int>& nums, int start, int length) {
        int middle = length / 2;
        int i = 0;
        while(i < middle) {
            swap(nums[start + i], nums[start + (length - i - 1)]);
            i++;
        }
    }

    void swap(int &a, int &b) {
        a ^= b;
        b ^= a;
        a ^= b;
    }