题目:给定一个数组,将数组中的元素向右移动 k
个位置,其中 k
是非负数。
例:
输入:nums = [1,2,3,4,5,6,7], k = 3
输出:[5,6,7,1,2,3,4]
解:
可以先反转整个数组,然后再反转前半部分,再反转后半部分
代码如下:
class Solution {
public void rotate(int[] nums, int k) {
int length = nums.length;
k %= length;
// 先反转整个数组
reverse(nums, 0, length - 1);
// 反转前半部分
reverse(nums, 0, k - 1);
// 反转后半部分
reverse(nums, k, length - 1);
}
public static void reverse(int nums[], int start, int end){
while(start < end){
int temp = nums[start];
nums[start++] = nums[end];
nums[end--] = temp;
}
}
}