算法入门-轮转数组

学习算法的第二天写的第二题，用了两种方法，下面是题目和链接：

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

示例 :
输入: 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]
来源：力扣（LeetCode）
链接： https://leetcode.cn/problems/rotate-array

第一种方法：我想的是直接建立一个新数组，将原数组里的元素插入进去即可，当然插入也分两种情况，下面是示意图：

下面是代码具体实现：

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

第二种方法：写完以后去逛评论区，发现了大师的三次翻转做法，首先写好翻转函数，假设原数组为123456，k=3，然后第一次将数组全部翻转，123456变为654321，第二次将第一次翻转的数组从0到k-1进行翻转(处理原坐标+k>=数组长度部分)，654321变为456321，最后一次将数组从k到nums.size()-1进行翻转(处理其余部分)，456321变为456123，下面是具体代码实现：

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