给定一个数组,将数组中的元素向右移动 k 个位置,其中 k 是非负数。
示例 1:
输入: [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:
输入: [-1,-100,3,99] 和 k = 2
输出: [3,99,-1,-100]
解释:
向右旋转 1 步: [99,-1,-100,3]
向右旋转 2 步: [3,99,-1,-100]
说明:
尽可能想出更多的解决方案,至少有三种不同的方法可以解决这个问题。
要求使用空间复杂度为 O(1) 的 原地 算法。
C++
暴力法,即每次向后移动一位,一共移动k次,一般会超时
class Solution {
public:
void rotate(vector<int>& nums, int k) {
int tmp;
int len = nums.size();
for(int j =0; j<k;j++)
{
tmp = nums[len-1];
for (int i=len-1;i>0;i--)
{
nums[i]=nums[i-1];
}
nums[0]=tmp;
}
}
};
2
跳跃法,
直接跳K步,
然后被替换的数在跳K步
如果回到初始位置,走到下一个数继续跳
class Solution {
public:
void rotate(vector<int>& nums, int k) {
k = k%nums.size();
int count=0;
int len = nums.size();
for (int i=0; count<len; i++)
{
int current = i;
int prev = nums[i];
do{
int next = (current+k)%len;
int tmp=nums[next];
nums[next]=prev;
prev = tmp;
current = next;
count++;
}
while(current!=i);
}
}
};
3
反转法:
移动K次,有K个数,从最后到之前,然后继续反转
原始数组 : 1 2 3 4 5 6 7
反转所有数字后 : 7 6 5 4 3 2 1
反转前 k 个数字后 : 5 6 7 4 3 2 1
反转后 n-k 个数字后 : 5 6 7 1 2 3 4 --> 结果
```cpp
class Solution {
public:
void rotate(vector<int>& nums, int k) {
int len = nums.size();
k=k%len;
reverse1(nums,0,len-1);
reverse1(nums,0,k-1);
reverse1(nums,k,len-1);
}
void reverse1(vector<int>& nums, int start, int end)
{
while(start<end)
{
int tmp = nums[start];
nums[start]=nums[end];
nums[end]=tmp;
start++;
end--;
}
}
};