LeetCode：189 旋转数组 O(n)翻转解法

给定一个数组，将数组中的元素向右移动 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]

说明:

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

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

思路

一开始想的是先写向右一次的函数once，然后调用k次once，就移动k次了，复杂度是O(n*k)，直接超时。。。

后面通过观察评论区 发现有一种骚解法，复杂度O(n)：

设数组长度为len，需要右移k次

• 翻转前 len 个元素
• 翻转整个数组
• 翻转前 len-k 个元素

解释：
原数组：

[1,2,3,4,5,6,7]

1.翻转前 len 个

[4,3,2,1,5,6,7]

2.整个翻转：

[7,6,5,1,2,3,4]

3.翻转前 len-k 个

[5,6,7,1,2,3,4]

代码

k 记得模 len，大大减少不必要的边界判断

class Solution
{
public:
    void rotate(vector<int>& nums, int k)
    {
        int len = nums.size();
        k %= len;
        reverse(nums.begin(), nums.begin()+len-k);
        reverse(nums.begin(), nums.end());
        reverse(nums.begin(), nums.begin()+k);
    }
};

O(n)还是很恐怖的。。。