题目
给定一个数组,将数组中的元素向右移动 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) 的原地算法。
解法1
分析过程
采用翻转数组的方法
1)翻转数组[0,len-k-1]范围的元素,示例1数组变为:4,3,2,1,5,6,7
2)翻转数组[len-k,len-1]范围的元素,数组变为:4,3,2,1,7,6,5
3)翻转数组[0,len-1]范围的元素,数组变为:5,6,7,1,2,3,4
代码实现
class Solution {
public void rotate(int[] nums, int k) {
int len=nums.length;
//右移位置k超过数组长度时,k取余
if(k>=len){
k=k%len;
}
reverse(nums,0,len-k-1);
reverse(nums,len-k,len-1);
reverse(nums,0,len-1);
}
public void reverse(int[] nums,int start,int end){
while(start<end){
int temp=nums[start];
nums[start]=nums[end];
nums[end]=temp;
start++;
end--;
}
}
}
解法2
高位移动
分析过程
1)定义标记cur,取出的高位元素放置的位置;循环k
2)将第len-k位置的元素取出赋值给temp,移动数组[cur,len-k-1]范围的元素均向右移动一位;移动完成后,将temp放在cur位置;
3)cur数值加1,k数值减1;循环执行步骤2,3,直至k=0,结束
示例过程
k=3;c=0;i:[3,0] 平移后数组:5,1,2,3,4,6,7
k=2,c=1;i:[4,1] 平移后数据:5,6,1,2,3,4,7
k=1,c=2,i:[5,2] 平移后数组:5,6,7,1,2,3,4
代码实现
class Solution {
public void rotate(int[] nums, int k) {
int len=nums.length;
int cur=0;
if(k>=len){
k=k%len;
}
while(k>0){
int temp=nums[len-k];
//循环右移
for(int i=len-k-1;i>=cur;i--){
nums[i+1]=nums[i];
}
nums[cur]=temp;
cur++;
k--;
}
}
}
解法3
创建新数组,空间复杂度O(n),不满足要求
分析过程
分析位置规律,旋转后,i位置的元素放在了(i+k)%len位置
代码实现
class Solution {
public void rotate(int[] nums, int k) {
int len=nums.length;
int[] results=new int[len];
for(int i=0;i<len;i++){
results[(i+k)%len]=nums[i];
}
for(int i=0;i<len;i++){
nums[i]=results[i];
}
}
}