LeetCode 189 旋转数组

题目

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

示例 1：

输入: [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：

输入: [-1,-100,3,99] 和 k = 2

输出: [3,99,-1,-100]

解释:

向右旋转 1 步: [99,-1,-100,3]

向右旋转 2 步: [3,99,-1,-100]

说明:

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/rotate-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

分析

易知，旋转 k 步等效于将 i 处元素移动到 (i+k)%n 处，n 为数组长。可使用完成移动的元素数作为终止。

注意：当 n%k == 0 时，数组被分为 n/k 组，会在第一组中循环重复，所以当重入该组时要偏移修正

代码

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        int n = nums.size();
        k = k % n;
        if (n <= 1 || k == 0) return;
        int start = 0, pri = nums[0], nxt;
        for (int i = 0, cnt = 0; cnt < n; cnt++) {
            i = (i + k) % n;
            nxt = nums[i];
            nums[i] = pri;
            if (i == start) { // 偏移修正
                i++, start++;
                pri = nums[i];
            } else {
                pri = nxt;
            }
        }
    }
};