leetcode-数组-简单-旋转数组

题目

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

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

输入: [-1,-100,3,99] 和 k = 2
输出: [3,99,-1,-100]
解释: 
向右旋转 1 步: [99,-1,-100,3]
向右旋转 2 步: [3,99,-1,-100]

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

解答

方法一：

首先想到的就是穷举操作每一个元素的的方法。

分析如下：

• 移动次数和数组长度相等，相当于没有移动。因此对于K大于数组长度的情况需要进行求余操作。
• 每次记录最后一个元素，然后进行移动操作。

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        if (nums.empty() || nums.size() == 1 ) {
            return;
        }
        if (nums.size() < k) {
            k = k % nums.size();
        }
        for (int i = 0; i < k; i++) {
            int last_ele = nums.back();
            ///数据后退的一个操作，依次后退 
            for (int j = nums.size() - 1; j >= 1; --j) {
                nums[j] = nums[j - 1];
            }
            nums[0] = last_ele;
        }
    }
};

这个方案在leetcode提交之后是运行超时的。 

方法二：

因为第一个办法不行了，所以继续想。

• 其实就是把数组分成两个部分，然后把两个部分进行交换，后边的换到前边。
• 后边的怎么换到前边呢，翻转一下就到前边了，但是这样两个小部分的顺序也是反的。
• 那针对两个小部分各自做一次反转就可以了。

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        if (nums.empty() ||nums.size() == 1 ) {
            return;
        }
        if (nums.size() < k) {
            k = k % nums.size();
        }
        std::reverse(nums.begin(), nums.end());
        std::reverse(nums.begin(), nums.begin() + k);
        std::reverse(nums.begin() + k, nums.end());
    }
};