给定一个整数数组 nums
,将数组中的元素向右轮转 k
个位置,其中 k
是非负数
实现方法1:每次移动一位,直到k个位置。
void roate1(vector<int>& nums, int k)
{
if(k <= 0)
{
return;
}
int len = nums.size();
int stride = k % len;
for(int i=0; i<stride; ++i)
{
int end = nums[len -1];
for(int j=len-1; j>0; --j)
{
nums[j] = nums[j-1];
}
nums[0] = end;
}
}
实现方法2: 旋转数组
#include <iostream>
#include <utility>
#include <vector>
void rotate(vector<int>& nums, int k)
{
int len = nums.size();
int stride = k % len;
int i=0;
while(i < len && stride != 0)
{
for(int j=0; j<stride && i<len; ++j)
{
std::swap(nums[i], nums[len-stride + j]);
++i;
}
if(len>i && len - i < stride)
{
stride = stride % (len -i);
}
}
}
int main(int argc, char** argv)
{
std::vector<int> arr{1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27};
rotate(arr, 38);
std::copy(arr.begin(), arr.end(), std::ostream_iterator<int>(cout," "));
std::cout<<std::endl;
return 0;
}
结果:
17 18 19 20 21 22 23 24 25 26 27 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16