问题描述:
给定一个数组,将数组中的元素向右移动 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) 的 原地 算法。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/rotate-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
执行结果:
代码描述:
- 方法一:
重新复制一个数组 temp, 然后把 temp 根据 k 值分两段,每段分别复制到原来的nums 数组中。
时间复杂度O(n), 空间复杂度O(n)。以下代码,不满足空间复杂度为1的要求。
class Solution {
public:
void rotate(vector<int>& nums, int k) {
if(nums.size() <= 1) return;
vector<int> temp(nums);
if(k >= nums.size())
{
k %= nums.size();
if(k == 0)
return;
}
for(int i = k, j = 0; i < nums.size(); ++i, ++j)
{
nums[i] = temp[j];
}
for(int i = 0, j = nums.size()-k; j < temp.size(); ++i, ++j)
{
nums[i] = temp[j];
}
}
};
- 方法二:
定一个旋转函数,先整体旋转,然后 1 --(k-1)旋转,然后(k-1)--(size-1)进行旋转,就得出了正确结果。
时间复杂度O(n), 空间复杂度O(1)。满足题目要求。
class Solution {
public:
void rotate(vector<int>& nums, int k) {
if(nums.size() <= 1) return;
if(k > nums.size())
{
k %= nums.size();
if(k == 0)
return;
}
reverse(nums, 0, nums.size()-1);
reverse(nums, 0,k-1);
reverse(nums, k, nums.size()-1);
}
void reverse(vector<int> &nums, int start, int end)
{
int temp = 0;
while(start < end)
{
temp = nums[start];
nums[start] = nums[end];
nums[end] = temp;
++start;
--end;
}
}
};