题目描述:
给定一个整数数组 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 rotate(vector<int>& nums, int k) {
int len = nums.size();
vector<int>newnum(len);
for(int n =0;n<len;n++){
newnum[(n+k)%len]=nums[n];
}
nums.assign(newnum.begin(),newnum.end());
}
};
方法二:采取总翻转后再分批翻转的方法
class Solution {
public:
void rotate(vector<int>& nums, int k) {
int len = nums.size();
// 防止k大于数组长度,取k的模
k %= len;
// 第一次翻转:整个数组翻转
reverse(nums.begin(), nums.end());
// 第二次翻转:前k个元素翻转
reverse(nums.begin(), nums.begin() + k);
// 第三次翻转:剩余元素翻转
reverse(nums.begin() + k, nums.end());
}
};
解析:
假设我们有以下数组:
nums = [1, 2, 3, 4, 5, 6, 7]
现在,我们要将这个数组向右旋转3个位置。我们可以使用代码中的 rotate
函数来实现它。
首先,我们计算 k
对数组长度的模,即 k % len
,这里 k = 3
,len = 7
,所以 k
对 len
的模是3。
1.接下来,我们进行第一次翻转,整个数组翻转,数组变为:
[7, 6, 5, 4, 3, 2, 1]
2.然后,我们进行第二次翻转,翻转前k个元素,也就是前3个元素:
[5, 6, 7, 4, 3, 2, 1]
3.最后,我们进行第三次翻转,翻转剩余的元素:
[5, 6, 7, 1, 2, 3, 4]