题目
给定一个数组,将数组中的元素向右移动 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) 的原地算法。
代码
我觉得这道题绝对不能使用数组元素后移或者前移这种方法,因为数组的移位操作效率非常的低。所以我决定使用替换的方式
代码写的比较粗糙:
public class Solution {
public void Rotate(int[] nums, int k) {
k=k%nums.Length;//当k大于数组长度的时候,就会出现数组旋转一圈又回到原点,这个过程简直是浪费资源,所以我们取余
int []a=new int[nums.Length-k];//创建一个数组来存放要替换的元素
for(int i=0;i<nums.Length-k;i++)
{
a[i]=nums[i];//将要替换的元素压入空数组
}
for(int i=nums.Length-k,index=0;i<nums.Length;i++,index++)
{
nums[index]=nums[i];//从0开始,将旋转的元素依次替换
}
for(int i=k,j=0;i<nums.Length;i++,j++)
{
nums[i]=a[j];
}
}
}
大佬的代码,思路比较巧妙:
public class Solution {
public void Rotate(int[] nums, int k)
{
k %= nums.Length;
Reverse(nums, 0, nums.Length-1);
Reverse(nums, 0, k - 1);
Reverse(nums, k, nums.Length-1);
}
public void Reverse(int[] nums, int start, int end)
{
while (start < end)
{
int t = nums[end];
nums[end] = nums[start];
nums[start] = t;
start++;
end--;
}
}
}