序言
该算法用于一维数组的移动。
题目描述
给定一个数组,将数组中的元素向右移动 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) 的 原地 算法。
算法分析
图片来源于LeetCode官方解答
1.假设从第一个nums【0】开始处理,设start=j=0,则nums【j】移动到nums【j+k】的位置,nums【j+k】的值放入到tempnext中,再让j+k,即开始位置变化。
2.把j+k看做下一次操作的起点,回到步骤一,直到j+k==start,说明现在构成了一轮循环。(每一次操作:j+k位的数保存起来,前一个数移动到第j+k位,j表示当前操作位)
3.把开始的结点往前挪一位,开始下一轮循环,防止因为numsSize%k==0的情况。
代码
void rotate(int* nums, int numsSize, int k){
int tempfront,tempnext,j,start=0;
k=k%numsSize;//防止下标溢出
tempfront=nums[0];
j=k;
for(int i=0;i<numsSize;i++){//因为每次处理一个数,所以循环numsSize后一定让数组达标。
if(start==j)//说明一轮循环结束,初始化tempfront和修改j与start
{
nums[j]=tempfront;//将上一轮循环的最后一个值放到正确位置
tempfront=nums[(j+1)%numsSize];
start=(j+1)%numsSize;
j=(j+k+1)%numsSize;
}else{//开始一次操作,每次让一个数在正确位置
tempnext=nums[j];
nums[j]=tempfront;
j=(j+k)%numsSize;
tempfront=tempnext;
}
}
}
分析代码可得时间复杂度O(n),空间复杂度(n)。