给定一个整数数组 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]
代码如下:
//反转数组的方法
class Solution {
public:
void reverse(vector<int>& nums,int left,int right)
{
while(left<=right)//反转数组的函数,数组进行前后反转
{
swap(nums[left],nums[right]);
left++;
right--;
}
}
void rotate(vector<int>& nums, int k) {
int n=nums.size()-1;//反转数组时,下标要在数组元素的最后一个
k%=nums.size();//k有可能大于数组长度,取余后使k的值小于数组长度
reverse(nums,0,n);//首先反转整个数组
reverse(nums,0,k-1);//然后反转数组[0,k-1]
reverse(nums,k,n);//再反转[k,n]
}
};