Rotate Array 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.

设 数组长度为n,移动步数为k

思路整理:

1.是否能通过公式计算出移动后的位置呢?

解决:假设当前位置为 a0,则移动后的位置为 a1 = (a0 + k) % n;

2.如果我们让nums[a1] = nums[a0],那么nums[a1]该怎么处理?

解决:可以计算a1位置移动后的下一个位置a2 = (a1 + k) % n,然后让nums[a2] = nums[a1],依次类推,直到下一个位置an回到起点位置 也就是 a,这样循环下去,总会将所有元素替换掉吧。

3.我试着编写出一个循环程序,提交后发现程序存在严重漏洞,如果 n % k = 0 或者 k % n = 0,我们执行这个程序会发现不能移动所有元素,比如 n = 6,k = 3,执行的操作是a0->a3,a3->a0,我们会发现a1,a2...等元素并没有替换掉。那么该怎么处理这个问题呢?

解决:抓住n % k = 0 或者 k % n = 0这种情况做分析,我发现可以加入一层外层循环,还是上面例子,外层循环是 a0,a1,a2,这样就可以保证所有元素可以替换掉了。

4.外层循环的执行次数 c 该如何限制呢?

解决:我使用了归纳假设方法,首先假设n % k = 0,c = k;k % n = 0,c = n,其他情况的话 c = 1,这种假设经过程序测试,发现是错误的,假如n = 6, k = 4很明显还是不能遍历所有元素。然后我继续使用归纳方法,举出很多种n和k的组合,最后大胆的假设c是n和k的最大公约数,事实证明,这是正确的。也许这个结论证明并不严谨,但已经足够解决这个问题了。

写出程序如下:

void rotate(vector<int>& nums, int k) {
        int n = nums.size();
        if (n == 0 || k == 0)
            return;
        // 求最大公约数
        int a = n, c = k, b = a % c;
        while (b != 0)
        {
            a = c;
            c = b;
            b = a % c;
        }
        // 依次替换
        while (c--)
        {
            int oldNum = nums[c];
            int curPos = (c + k) % n;
            while (curPos != c)
            {
                int curNum = nums[curPos];
                nums[curPos] = oldNum;
                curPos = (curPos + k) % n;
                oldNum = curNum;
            }
            nums[curPos] = oldNum;
        }
    }

这里要注意依次替换的时候,我们要使用一个临时变量oldNum来保存上一次被替换掉的元素。循环体内的赋值过程有些绕,要理清思绪。我在这里卡了一阵子,越想越头晕。。。

算法总结:

这个算法成功的将时间复杂度控制在O(n),而且空间复杂度是O(1),更可贵的是所有操作没有改变内存位置。可以说是非常理想了。

查看了下网上的其他方法,大体上有三种,

一种是将所有元素从后往前依次移动一个位置,然后外层循环k次。这种方法时间复杂度为O(n*k),不消说,没有多少效率。

代表算法(C语言实现):

void rotate(int nums[], int n, int k) {
    int temp;
    for (int step = 0; step < k; step++) {
        temp = nums[n-1];
        for (int i = n-1; i > 0; --i)
        {
            nums[i] = nums[i-1];
        }
        nums[0] = temp;
    }
}

一种是反转数组的方法,先把整个数组reverse,然后把前面的reverse回来,再把后面的reverse回来,代表算法:

void rotate(int nums[], int n, int k) {
    k = k % n;
    reverse(nums, nums + n);
    reverse(nums, nums + k);
    reverse(nums + k, nums + n);
}

这个算法比较高明,时间复杂度可以控制在O(n),空间复杂度也是O(1),非常理想,但要看你的reverse实现是否高效

还有一种方法个人认为比较投机取巧,它使用C语言的内存操作函数,改变了内存地址,有一定风险,但算法确实高效

void rotate(int nums[], int n, int k) {
    k = k % n;
    if (k == 0) return;
    int *temp = new int[n];
    memcpy(temp, nums+(n-k), sizeof(int)*k);
    memcpy(temp+k, nums, sizeof(int)*(n-k));
    memcpy(nums, temp, sizeof(int)*n);
    delete[] temp;
}

算法——让人类思维更加理性

原文地址:https://www.cnblogs.com/sdlwlxf/p/5095013.html