题目描述(传送门)
给定一个数组,将数组中的元素向右移动 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) 的 原地 算法。
解题思路
这道题是一维数组的旋转,之前在二维数组旋转的时候提到过,可以参考【每日一题】力扣48.旋转图像
思路:对于一维数组向右移动 k 个位置,可以将前n-k-1个翻转,后n-k翻转再将整体翻转。就可以原地实现一维数组旋转。(具体边界根据题目如何定义下标来定)。
注意:当数组元素小于等于1时直接返回即可,当k的值大于数组长度则只需移动k%length;
Java代码
class Solution {
public void rotateHelper(int[] nums, int left, int right) {
while (left < right) {
int tmp = nums[left];
nums[left] = nums[right];
nums[right] = tmp;
left++;
right--;
}
}
public void rotate(int[] nums, int k) {
if (nums.length <= 1) {
return;
}
k = k%nums.length;
rotateHelper(nums, 0, nums.length-k-1);
rotateHelper(nums, nums.length-k, nums.length-1);
rotateHelper(nums, 0, nums.length-1);
}
}
C语言代码
void reverse(int* nums,int start,int end)
{
for(int j=0;start<end;j++)
{
int tmp=nums[end];
nums[end]=nums[start];
nums[start]=tmp;
start++;
end--;
}
}
void rotate(int* nums, int numsSize, int k)
{
reverse(nums,0,numsSize-1);
if(k>numsSize)
{
k%=numsSize;
}
reverse(nums,0,k-1);
reverse(nums,k,numsSize-1);
}