给你一个数组,将数组中的元素向右轮转 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& 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()); //逆转整个数组
}
};