给定一个整数数组 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