旋转数组
给定一个数组,将数组的元素向右移动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]
思路1:旋转k次
//时间复杂度:O(N*K)
void rotate(int* nums, int numsSize, int k) {
while (k--) {
int tmp = nums[numsSize - 1];
for (int end = numsSize - 2; end >= 0; --end) {
nums[end + 1] = nums[end];
}
nums[0] = tmp;
}
}
思路2:以空间换时间
开一个跟它一样大的数组,把后k个拷贝到前面,把前numsSize-k拷贝到后面
时间复杂度:O(N)
(最妙的)思路3:
后k个逆置,前n-k个逆置,再整体逆置
例: [1,2,3,4,5,6,7]
后k个逆置 [1,2,3,4,7,6,5]
前n-k个逆置 [4,3,2,1,7,6,5]
再整体逆置 [5,6,7,1,2,3,4]
void Reverse(int* nums, int left, int right) {
while (left<right)
{
int tmp = nums[left];
nums[left] = nums[right];
nums[right] = tmp;
++left;
--right;
}
}
void rotate(int* nums, int numsSize, int k) {
if (k >= numsSize) {
k %= numsSize;
}
Reverse(nums, numsSize - k, numsSize - 1);
Reverse(nums, 0, numsSize - k - 1);
Reverse(nums, 0, numsSize - 1);
}