旋转数组
题目描述
给定一个数组,将数组中的元素向右移动 k 个位置,其中 k 是非负数。
使用空间复杂度为 O(1) 的 原地 算法解决
示例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]
使用语言: C语言
思路解析: 因为空间复杂度为O(1)所以只能使用原地算法,仔细观察会发现,不论旋转几次,只需要逆置三次即可完成, 先将整个数组分为[0 , numsSize - k - 1] [numsSize - k , numsSize - 1]两个区间然后分别进行逆置, 最后在对整个区间进行逆置[0 , numsSize - 1]
图解如下:
需要注意的是:问题:如果k>numsSize第一步的逆置区间边界会减到负数
解决方法:如果逆置numsSize次相当于没有逆置,又回到了原处,所以当
K >= numsSize时可以k %= numsSize
代码实现
//逆置
void Reverse (int* nums,int left,int right)
{
while(left < right)
{
int tem = nums[left];
nums[left] = nums[right];
nums[right] = tem;
left++;
right--;
}
}
void rotate(int* nums, int numsSize, int k)
{
//如果k>numsSize下面的逆置区间会减到负数,如果逆置numsSize次相当于没有逆置,
//使用当K>=numsSize时可以k%numsSize
if(k >= numsSize)
{
k %= numsSize;
}
//进行三次逆置
Reverse (nums , 0 , numsSize - k - 1);
Reverse (nums , numsSize - k , numsSize - 1);
Reverse (nums , 0 , numsSize - 1);
}