算法二十天
第十天
旋转数组
给你一个数组，将数组中的元素向右轮转 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]提示：
1 <= nums.length <= 10^5
-231 <= nums[i] <= 231 - 1
0 <= k <= 105
进阶：
尽可能想出更多的解决方案，至少有 三种 不同的方法可以解决这个问题。
你可以使用空间复杂度为 O(1) 的 原地 算法解决这个问题吗？
作者：力扣 (LeetCode)
链接：https://leetcode-cn.com/leetbook/read/top-interview-questions-easy/x2skh7/
法一：将数组中的数字一个一个交换到合适位置，时间复杂度O(n^2),空间复杂度O(1)。
#include<bits/stdc++.h>
using namespace std;
void rotate(vector<int>& nums, int k) {
int first=nums.size()-(k%nums.size());//移动后数组是以这个数字作为新数组的开头
for(int i=first;i<nums.size();i++)
{
for(int j=i;j>i-first;j--)//将first后的数组向前移动对应位置
{
int temp=nums[j];
nums[j]=nums[j-1];
nums[j-1]=temp;
}
}
}
int main()
{
vector<int> prices={
1,2,3,4,5,6,7};
rotate(prices,3);
for(auto& i:prices)
cout<<i<<"";
}
法二：建立新的数组，遍历一遍即可时间复杂度O(n),空间复杂度O(n)。
#include<bits/stdc++.h>
using namespace std;
void rotate(vector<int>& nums, int k) {
int first=nums.size()-(k%nums.size());//同上
if(first==0)
return;
vector<int> temp;
for(int i=first;i<nums.size();i++)
temp.push_back(nums[i]);
for(int i=0;i<first;i++)
temp.push_back(nums[i]);
nums=temp;
}
int main()
{
vector<int> prices={
1,2,3,4,5,6,7};
rotate(prices,3);
for(auto& i:prices)
cout<<i<<"";
}
法三：动态数组解决，时间复杂度O(n^2),空间复杂度O(1);
#include<bits/stdc++.h>
using namespace std;
void rotate(vector<int>& nums, int k) {
for(int i=0;i<k;i++)
{
nums.insert(nums.begin(),nums[nums.size()-1]);//临时的空间损耗不计入总的空间复杂度；
nums.pop_back();
}
}
int main()
{
vector<int> prices={
1,2,3,4,5,6,7,8};
rotate(prices,3);
for(auto& i:prices)
cout<<i<<"";
}
法四：找到位置直接交换，时间复杂度O(n),空间复杂度O(1);
#include<bits/stdc++.h>
using namespace std;
void rotate(vector<int>& nums, int k) {
int temp1;
k%=nums.size();
if(k==0||nums.size()==1)
return;
int temp=nums[0];
int count=0;
int sum=k;
for(int j=0;j<nums.size();j++)
{
temp1=nums[sum%nums.size()];
nums[sum%nums.size()]=temp;
temp=temp1;
if(sum%nums.size()==count)
{
count++;
sum=count;
temp=nums[sum%nums.size()];
}
sum+=k;
}
}
int main()
{
vector<int> prices={
1,2,3,4,5,6};
rotate(prices,4);
for(auto& i:prices)
cout<<i<<" ";
}