HOT15-旋转数组

       leetcode原题链接：旋转数组

       上一篇：HOT14-合并区间

       下一篇：HOT16-除自身以外数组的乘积

题目描述

       给定一个整数数组 nums，将数组中的元素向右轮转 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]

解题方法：先将整个数组逆转，再将[0, n - k -1]、[n - k, n -1]分别逆转，同时考虑n小于k的情况。

C++代码

#include <iostream>
#include <algorithm> // std::swap
class Solution {
public:
    void reverse(std::vector<int>& nums, int left, int right) {
        while (left < right) {
            std::swap(nums[left], nums[right]);
            left++;
            right--;
        }
    }
    void rotate(std::vector<int>& nums, int k) {
        int n = nums.size();
        k = k % n; // 处理k大于n的情况
        reverse(nums, 0, n - 1); // 整个数组翻转
        reverse(nums, 0, k - 1); // 翻转前k个部分
        reverse(nums, k, n - 1); //翻转后半部分
    }
};