给定一个数组,将数组中的元素向右移动 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]来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/rotate-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
这里,用了最简单的方法实现,复杂度为o(n)
思路是:利用一个新数组,填充后再赋值给原数组。最关键的就是:newArr[(k+i)%numsSize]=nums[i];
位置 p的元素会放至 (p+k) mod n的位置。利用取余,就可以实现旋转。
void rotate(int* nums, int numsSize, int k){
int newArr[numsSize];
for(int i=0;i<numsSize;i++){
newArr[(k+i)%numsSize]=nums[i];
}
for(int j=0;j<numsSize;j++){
nums[j]=newArr[j];
}
return nums;
}