力扣189-轮转数组-C++三行代码

一、运行结果

 

二、题目

给你一个数组，将数组中的元素向右轮转 k 个位置，其中 k 是非负数。

示例 1:

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

输入：nums = [-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可能比输入数组长度len要大，而多轮转len次相当于没有轮转，为确保轮转次数小于数组长度，先将k 对len进行求余操作。

我们将数组进行划分，将后面k个元素成为后一段，将前面剩余的len - k 个元素称为前一段，可以想象得到，轮转k次后，其实就相当于将后一段的k个元素直接插到原数组的头部，而前一段的每一个元素都相当于后移k个位置。 新数组的第一个元素就是原数组后一段的第一个元素，新数组的最后一个元素就是原数组前一段的最后一个元素（这里可以画图来帮助理解），所以我们可以进行三次逆转完成轮转。首先分别对原数组的前一段和后一段进行逆转，而后再将整个数组逆转，就得到了轮转k次的结果。

四、代码

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        int len = nums.size();
        k %= len;
        if(k == 0) return;
        reverse(nums.begin(), nums.begin() + len - k); //逆转前len-k个元素
        reverse(nums.begin() +len -k , nums.end()); //逆转后面k个元素
        reverse(nums.begin(), nums.end());  //逆转整个数组
    }
};