【leetcode】189. 旋转数组( Rotate Array )

---

题目描述

【leetcode】189. 旋转数组( Rotate Array )

给定一个数组，将数组中的元素向右移动 k 个位置，其中 k 是非负数。
说明:
尽可能想出更多的解决方案，至少有三种不同的方法可以解决这个问题。
要求使用空间复杂度为 O(1) 的 原地 算法。

第一次解答

思路：
题意是把数组nums向右平移k次，超出边界的部分从数组首插入。
简单的方法是按部就班的平移1次，2次，…，一直到k次，该方法缺点就是每次平移所有元素都要移动一次，复杂度是k*nums.size()。
另一种思路是一步计算得到每个元素最终所在的位置，将元素直接放到最终位置。用一个临时变量保存原位置值，但是这个代码不好实现，我写了几次都有bug，最后还是参考了官方题解。

test case:
[]
1
[1]
1
[1,2]
1
[1,2]
2
[1,2]
3
[1,2,3,4,5,6,7]
3
[1,2,3,4,5,6]
2

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        if(nums.size() == 0)
            return;
        k = k%nums.size(); // 若k>=nums.size()，平移k次与平移k%nums.size()次效果相同
        if(0 == k)
            return;
        
        int count = 0; // 记录已变换位置的元素个数
        // 注意这个for循环执行的判断条件
        // 若 nums.size() % k == 0， 则for循环执行k次，直到count==nums.size()
        // 若 nums.size() % k != 0， 则for循环执行1次（因为在do-while里一直执行到count==nums.size()）。
        for(int start=0; count<nums.size(); ++start){
            int current = start;
            int last_value = nums[current];
            int next;
            // 注意这里用do-while而不是wihle
            do{
                next = (current + k) % nums.size();
                int temp = nums[next];
                nums[next] = last_value;
                last_value = temp;
                current = next;
                ++count;
            }while(start != current);
        }
    }
};

结果：

截图

相关/参考链接