题目
LeetCode:189. 旋转数组
给定一个数组,将数组中的元素向右移动 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) 的 原地 算法。
代码
一
class Solution {
public:
void rotate(vector<int>& nums, int k) {
k=k%nums.size();
for(int i=0;i<nums.size()/2;i++)
{
int a=nums[i];
nums[i]=nums[nums.size()-1-i];
nums[nums.size()-1-i]=a;
}
for(int i=0;i<k/2;i++)
{
int a=nums[i];
nums[i]=nums[k-1-i];
nums[k-1-i]=a;
}
for(int i=k;i<(nums.size()+k)/2;i++)
{
int a=nums[i];
nums[i]=nums[nums.size()-1+k-i];
nums[nums.size()-1+k-i]=a;
}
}
};
二
class Solution {
public:
void rotate(vector<int>& nums, int k) {
k %= nums.size();
reverse(nums.begin(), nums.end());
reverse(nums.begin(), nums.begin() + k );
reverse(nums.begin() + k, nums.end());
// begin返回指向第一个元素的迭代器,
//end返回最后一个元素的下一个元素的迭代器
}
};
思路
反转3次,下次遇到能不能想到?
先把整个数组反转一次,再反转前K个数,最后反转K+1到最后的数。