给定一个数组,将数组中的元素向右移动 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) 的 原地 算法。
题目链接:https://leetcode-cn.com/problems/rotate-array/
思路
边界条件:当k>数组长度len时,移动的距离为k%len
如何移动:
1.新建数组用于临时存储,再复制回去
时间复杂度O(n),空间复杂度O(n)。
class Solution {
public:
void rotate(vector<int>& nums, int k) {
int len = nums.size();
if(len==0 || k%len==0) return;
k %= len;
vector<int> res(len);
for(int i=0; i<len; ++i){
res[(i+k)%len] = nums[i];
}
nums = res;
}
};
2.三次逆序
时间复杂度O(n),空间复杂度O(1)。
例子:1 2 3 4 5, k=2,所需结果为 4 5 1 2 3
第一次整个数组逆序:5 4 3 2 1
第二次[0, k%len-1]逆序: 4 5 3 2 1
第三次[k%len, len)逆序:4 5 1 2 3
class Solution {
public:
void revers(vector<int>& nums, int start, int end){
while(start<end){
int tmp = nums[start];
nums[start++] = nums[end];
nums[end--] = tmp;
}
}
void rotate(vector<int>& nums, int k) {
int len = nums.size();
if(len==0 || k%len==0) return;
k %= len;
revers(nums, 0, len-1);
revers(nums, 0, k-1);
revers(nums, k, len-1);
}
};