给定一个数组,将数组中的元素向右移动 k 个位置,其中 k 是非负数。
进阶:
尽可能想出更多的解决方案,至少有三种不同的方法可以解决这个问题。
你可以使用空间复杂度为 O(1) 的 原地 算法解决这个问题吗?
稍微看了眼题解写的,看题想到的就是第一种使用额外的数组来解,但是看进阶要求就达不到空间复杂度为O(1),
想了一会就去看题解怎么说了
class Solution {
public void rotate(int[] nums, int k) {
//看了题解,有三种解法,第一种使用额外的数组,第二种数学上的解法,第三种是旋转数组
k%=nums.length;
reverse(nums,0,nums.length-1);//全部逆转
reverse(nums,0,k-1);//逆转前k个
reverse(nums,k,nums.length-1); //逆转剩下的,得到输出
}
//逆转数组
public int[] reverse(int[] nums,int start,int end){
while(start<end){
int temp = nums[start];
nums[start] = nums[end];
nums[end] = temp;
start+=1;
end-=1;
}
return nums;
}
}
使用额外数组的解法
代码量少,但是时间和空间都不咋地
class Solution {
public void rotate(int[] nums, int k) {
int n = nums.length;
int[] newArr = new int[n];
for (int i = 0; i < n; ++i) {
newArr[(i + k) % n] = nums[i];
}
System.arraycopy(newArr, 0, nums, 0, n);
}
}