这个题的一种解法挺有意思,记录一下。
给定一个数组,将数组中的元素向右移动 k 个位置,其中 k 是非负数。
输入: [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]
研究了半天,发现一个规律,把数组分成两部分 a[1,2,3,4] b[5,6,7]
把a b两部分单独旋转为[4,3,2,1,7,6,5]
然后总体旋转[5,6,7,1,2,3,4]
得解~~
class Solution {
public:
void revert(vector<int>& nums, int start, int end){
while(start<end){
nums[start] ^= nums[end];
nums[end] ^= nums[start];
nums[start] ^= nums[end];
++start;
--end;
}
}
void rotate(vector<int>& nums, int k) {
int size=nums.size();
k %= size;
if(k==0)
return;
revert(nums,0,size-k-1);
revert(nums,size-k, size-1);
revert(nums,0, size-1);
}
};