.轮转数组

55 阅读1分钟

该题我们之前也是做过的,

我们可以使用额外的数组来将每个元素放至正确的位置。用 nn 表示数组的长度,我们遍历原数组,将原数组下标为 ii 的元素放至新数组下标为 (i+k)\bmod n(i+k)modn 的位置,最后将新数组拷贝至原数组即可

如下:

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
            vector<int> result(nums.size(),0);
            int n=nums.size();
            for(int i=0;i<n;i++)
            {
                result[(i+k)%n]=nums[i];
            }
            for(int i=0;i<n;i++)
            {
                nums[i]=result[i];
            }
    }
};

主要我们要介绍下面这种新方法——数组翻转

该方法基于如下的事实:当我们将数组的元素向右移动 k 次后,尾部  k mod n 个元素会移动至数组头部,其余元素向后移动  k mod n 个位置。

该方法为数组的翻转:我们可以先将所有元素翻转,这样尾部的k mod n 个元素就被移至数组头部,然后我们再翻转 [0, k\bmod n-1]区间的元素和 [k\bmod n, n-1]区间的元素即能得到最后的答案。

我们以 n=7,k=3 为例进行如下展示:

 

class Solution {
public:
    void reverse(vector<int>& nums, int start, int end) {
        while (start < end) {
            swap(nums[start], nums[end]);
            start += 1;
            end -= 1;
        }
    }

    void rotate(vector<int>& nums, int k) {
        k %= nums.size();
        reverse(nums, 0, nums.size() - 1);
        reverse(nums, 0, k - 1);
        reverse(nums, k, nums.size() - 1);
    }
};