关于长度为n的数组的循环右(左)移k个位置
循环右移k个位置
leetcode原题
给定一个数组,将数组中的元素向右移动 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)** 的 原地 算法。
采用三次旋转的方法
长度为n的数组循环右移:
1.把前n-k个元素逆置
2.把后k个元素逆置
3.把整个数组逆置
tips:这里也可以先逆置整个数组,在分别逆置前半部分和后半部分,不过应该改为
1.把整个数组逆置
2.把前n-k-1个元素逆置
3.把后k+1个元素逆置
长度为n的数组循环左移:
1.把前k个元素逆置
2.把后n-k个元素逆置
3.把整个数组逆置
tips:这里也可以先逆置整个数组,在分别逆置前半部分和后半部分,同上需要更改逆置的元素个数,此处略
代码
void reverse(int* nums,int low,int high)
{
int temp,i,j;
for(i = low,j = high;i < j;i++,j--)
{
temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
}
}
void rotate(int* nums, int numsSize, int k){
k%=numsSize; //这里取余防止k > n的情况
reverse(nums ,0 ,numsSize - k - 1);
reverse(nums ,numsSize - k ,numsSize - 1);
reverse(nums ,0 ,numsSize - 1);
}