轮转数组
- 题目描述
将已知的数组中的元素向右轮转 k 个位置。( k 是非负数)
输入(k=3) | 输出 |
---|---|
1234567 | 5671234 |
class Solution {
public:
void rotate(vector<int>& nums, int k) {
int n= nums.size();
vector<int>newArr;
int j = nums.size()-k;
while(k>n)
{k=k%n;}
for(int i = 1;i<=k;i++)
{
if(k<=0) {break;}
newArr.push_back(nums[j]);
j++;
}
for(int i = 0;i<nums.size()-k;i++)
{
newArr.push_back(nums[i]);
}
nums.assign(newArr.begin(), newArr.end());
}
};
这个是我自己的做法,结果是对的,但是超出了时间限制
利用新数组法:
class Solution {
public:
void rotate(vector<int>& nums, int k) {
int n = nums.size();
vector<int>new(n);
for(int i =0;i<n;i++)
{
new[(i+k)%n]=nums[i];
}
nums.assign(new.begin(),new.end());
}
};
这种方法时间复杂度和空间复杂度均为O(n)
** 翻转 **class Solution { public: void reverse(vector<int>& nums, int start, int end) {
翻转 | |
---|---|
1234567 | ①7654321 |
7654321 | ②5674321 |
5674321 | ③5671234 |
第一次
while(start<end)
{
swap(nums[start],nums[end]);
start++;end--;
}
while(k>n)
k%=n;
第二次
reverse(nums,0,k-1)
第三次
reverse(nums,k,n-1)