给定一个数组,将数组中的元素向右移动 k 个位置,其中 k 是非负数。
示例 1:
输入: [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:
输入: [-1,-100,3,99] 和 k = 2
输出: [3,99,-1,-100]
解释:
向右旋转 1 步: [99,-1,-100,3]
向右旋转 2 步: [3,99,-1,-100]
说明:
尽可能想出更多的解决方案,至少有三种不同的方法可以解决这个问题。
要求使用空间复杂度为 O(1) 的原地算法。
第一种:一个一个替换,比较繁琐。
class Solution {
public void rotate(int[] nums, int k) {
k = k % nums.length;
//每次往后移一位,移动k次
for (int i = 0; i < k; i++) {
int fir = nums[nums.length - 1];
//从倒数第二个元素开始移动
for (int j = nums.length - 2; j >= 0; j--) {
nums[j + 1] = nums[j];
}
//每次的第一个元素都是移动前的最后一个
nums[0] = fir;
}
}
}
第二种,将旧数组的元素按照替换后的位置重新赋给新数组
class Solution {
public void rotate(int[] nums, int k) {
k = k % nums.length;
int[] nnums = new int[nums.length];
for (int i = 0; i < nums.length; i++) {
int num = (i + k) % nums.length;
nnums[num] = nums[i];
}
for (int i = 0; i < nnums.length; i++) {
nums[i] = nnums[i];
}
}
}
第三种,比较难理解,博主也是通过debug一步一步才弄明白,不懂的可以去debug下,通过自己弄清楚。先是将整个数组全部反转,比如1,2,3,4变为4,3,2,1.之后,通过k值,决定将0到k-1个元素反转,再将k到最后一个反转,例如k为2时,将4,3反转,再将2,1反转。
class Solution {
public void rotate(int[] nums, int k) {
if(nums==null || nums.length==0){
return ;
}
if(nums.length==1){
return ;
}
k=k%nums.length;
reverse(nums);
reverse(nums,0,k-1);
reverse(nums,k,nums.length-1);
}
public void reverse(int[] arr,int begin,int end){
int half=(end-begin+1)/2;
int tar;
for(int i=0;i<half;i++){
tar=arr[begin];
arr[begin]=arr[end];
arr[end]=tar;
begin++;
end--;
}
}
public void reverse(int[] arr){
reverse(arr,0,arr.length-1);
}
}