题目
给你一个数组,将数组中的元素向右轮转 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
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);
}
在这里和大家分享一个错误,就是想原地解决,就直接nums[(i+k)%n]=nums[i];
直接将每个数字放至它最后的位置,这样之前被放置在这个位置的元素会被覆盖从而丢失。
方法二:数组翻转
根据题意,如果使用多余数组存储空间,会导致空间复杂度为 n,所以在这里,我们可以使用常量级的空间复杂度解法:数组翻转首先对整个数组实行翻转,这样子原数组中需要翻转的子数组,就会跑到数组最前面。这时候,从 k 处分隔数组,左右两数组,各自进行翻转即可
代码2
public void rotate1(int[] nums, int k) {
int n=nums.length;
k=k%n;
reverse(nums,0,n-1);
reverse(nums,0,k-1);
reverse(nums,k,n-1);
}
public void reverse(int []nums,int start,int end){
while(start<end){
int temp=nums[end];
nums[end]=nums[start];
nums[start]=temp;
start++;
end--;
}
}