旋转数组【力扣】

题目描述

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

题目分析

我的思路是采用翻转的方法，即：三次翻转

1. 将整个数组翻转
2. 翻转前k个
3. 翻转后k个

nums = [1,2,3,4,5,6,7], k = 3
7 6 5 4 3 2 1
5 6 7 4 3 2 1
5 6 7 1 2 3 4

代码实现

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        k = k % nums.size();// 让k的值范围保持在0 ~ nums.size()
        // reverse 参数范围 [first, last) ，左闭右开，
        reverse(&nums[0], &nums[nums.size()]);
        reverse(&nums[0], &nums[k]);
        reverse(&nums[k], &nums[nums.size()]);
    }
};