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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

执行结果：

代码描述：

• 方法一：

重新复制一个数组 temp， 然后把 temp 根据 k 值分两段，每段分别复制到原来的nums 数组中。

时间复杂度O(n), 空间复杂度O(n)。以下代码，不满足空间复杂度为1的要求。

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        if(nums.size() <= 1)    return;
        vector<int> temp(nums);
        if(k >= nums.size())
        {
            k %= nums.size();
            if(k == 0)
                return;
        } 
        for(int i = k, j = 0; i < nums.size(); ++i, ++j)
        {
            nums[i] = temp[j];
        }
        for(int i = 0, j = nums.size()-k; j < temp.size(); ++i, ++j)
        {
            nums[i] = temp[j];
        }
    }
};

• 方法二：

定一个旋转函数，先整体旋转，然后 1 --（k-1）旋转，然后（k-1）--（size-1）进行旋转，就得出了正确结果。

时间复杂度O(n)， 空间复杂度O(1)。满足题目要求。

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        if(nums.size() <= 1)    return;
        if(k > nums.size())
        {
            k %= nums.size();
            if(k == 0)
                return;
        }
        reverse(nums, 0, nums.size()-1);
        reverse(nums, 0,k-1);
        reverse(nums, k, nums.size()-1);
            
    }
    void reverse(vector<int> &nums, int start, int end)
    {
        int temp = 0;
        while(start < end)
        {
            temp = nums[start];
            nums[start] = nums[end];
            nums[end] = temp;
            ++start;
            --end;
        }
    }
};