初级算法-旋转数组

旋转数组

题目

给定一个数组，将数组中的元素向右移动 k 个位置，其中 k 是非负数。

进阶：

尽可能想出更多的解决方案，至少有三种不同的方法可以解决这个问题。
你可以使用空间复杂度为 O(1) 的 原地 算法解决这个问题吗？

示例 1:

输入: nums = [1,2,3,4,5,6,7], k = 3
输出: [5,6,7,1,2,3,4]
解释:
向右旋转 1 步: [7,1,2,3,4,5,6]
向右旋转 2 步: [6,7,1,2,3,4,5]
向右旋转 3 步: [5,6,7,1,2,3,4]
示例 2:

输入：nums = [-1,-100,3,99], k = 2
输出：[3,99,-1,-100]
解释:
向右旋转 1 步: [99,-1,-100,3]
向右旋转 2 步: [3,99,-1,-100]

提示：

1 <= nums.length <= 2 * 104
-231 <= nums[i] <= 231 - 1
0 <= k <= 105

分析：

方案一
构建一个等大的数组，遍历原数组并将遍历后的数据储存到新数组中，最后复制到原数组中。

方案二
原地算法尝试
思路：从第一个元素开始，向右移动k个单位，再将第k个元素向右移动k个单位，以此类推。
如果数组长度n和k互质，则会共移动n*k（最小公倍数）个单位后遍历全部数组；
如果n和k不互质，求两者最大公约数，则第一个元素会移动最小公倍数个单位后回到原地，未遍历完全。这时开始移动第二个元素，直到移动完最大公约数个元素。

代码：

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        //原地算法尝试
        //求num长度n和k的最大公约数
        //如果两数互质，则抓住一个元素不断向右移动k个单位可以遍历整个数组
        //如果不互质，则要抓住最大公约数个元素以遍历整个数组
        int n=nums.size(),tem1=0,tem2=0,divisor=0;
        int i=0,j=0;
        if(k!=0){
            if(n>k)divisor=fun(n,k);
            else divisor=fun(k,n);
            //不互质
            if(divisor != 1){
               for(i=0;i < divisor;++i){
                    for(j=i,tem1=nums[i%n];(j+k)%n != i;j+=k){
                        tem2=nums[(j+k)%n];
                        nums[(j+k)%n]=tem1;
                        tem1=tem2;
                    }
                nums[i]=tem1;
                }
            }
            //互质
            else{
                for(i=0,tem1=nums[i%n];(i+k)%n !=0;i+=k){
                    tem2=nums[(i+k)%n];
                    nums[(i+k)%n]=tem1;
                    tem1=tem2;
                }
                nums[0]=tem1;
            }
        }
    }
    //辗转相除法
    int fun(int m,int n){
	    if(n==0) return m;
	    return fun(n,m%n);
    }
};

题解分析：

方法——环状替换
基本思路同分析类似，代码书写上有很多简略之处，值得借鉴。

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        int n = nums.size();
        //k如果大于n，必然被mod除，减少计算量
        k = k % n;
        //调用库函数gcd，也避免了k为0的情况
        int count = gcd(k, n);
        //count为1的情况也包含在内
        for (int start = 0; start < count; ++start) {
        //用current和prev含义更明晰
            int current = start;
            int prev = nums[start];
            //用do和while防止开始便不能进入循环的情况
            do {
                int next = (current + k) % n;
                //交换数值
                swap(nums[next], prev);
                current = next;
            } while (start != current);
        }
    }
};

方法——数组翻转

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

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

操作	结果
原始数组	1 2 3 4 5 6 7
翻转所有元素	7 6 5 4 3 2 1
翻转[0,k mod n-1]	5 6 7 4 3 2 1
翻转[k mod n,n-1]	5 6 7 1 2 3 4

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);
    }
};

题目来源于力扣