给你一个数组,将数组中的元素向右轮转 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]
提示:
1 <= nums.length <= 105
-231 <= nums[i] <= 231 - 1
0 <= k <= 105
思路:
1、用k对nums.length取模得到每个数字需要向前移动的步数m;
2、将整个数组翻转
3、将前m个数字和后nums.length-m个数字分别翻转
//翻转函数
void reserve(int* nums,int numsSize){
int* pre=nums+numsSize-1;
while(pre>nums){
int temp=*pre;
*pre=*nums;
*nums=temp;
pre--;
nums++;
}
}
//目标函数
void rotate(int* nums, int numsSize, int k){
int num=k%numsSize;
reserve(nums,numsSize);
reserve(nums,num);
reserve(&nums[num],numsSize-num);
}
时间复杂度:O(n);
空间复杂度:O(1);
翻转数组的解法具备较低时空复杂度和代码简洁度,但是还有更加优秀的数组轮换的方法,常数时间远远低于翻转数组的解法。