题目描述
给定一个整数数组 nums,将数组中的元素向右轮转 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]
示例 2:
输入:nums = [-1,-100,3,99], k = 2
输出:[3,99,-1,-100]
解释:
向右轮转 1 步: [99,-1,-100,3]
向右轮转 2 步: [3,99,-1,-100]
提示:
1 <= nums.length <= 105
-231 <= nums[i] <= 231 - 1
0 <= k <= 105
代码
递归:
int move(int* nums, int numsSize, int origin, int k, int n, int target){if(target == origin){nums[origin] = n;return 1;}int tmp = nums[target];nums[target] = n;n = tmp;return move(nums, numsSize, origin, k, n, (target+k)%numsSize)+1;
}
void rotate(int* nums, int numsSize, int k) {int i = 0;int cnt = 0;int n = 0;while(cnt != numsSize){n = nums[i];cnt += move(nums, numsSize, i, k, n, (k+i)%numsSize);i++;}
}
迭代:
void rotate(int* nums, int numsSize, int k) {int idex = 0;int cnt = 0;int n = 0;// 移动次数判断while(cnt != numsSize){// idex为移动起点n = nums[idex];int target = (idex+k)%numsSize;// 移动回到起点while(target != idex){int tmp = nums[target];// 移动nums[target] = n;n = tmp;target = (target+k)%numsSize;cnt++;}nums[target] = n;cnt++;idex++;}
}
代码思路:记录移动前的数值,移动,找到下一次移动的位置。
注意:一次移动可能不会完成整个数组移动,需要修改起点继续移动完成。
总结
扩展:修改代码中的加减,右移可以改成左移。