189. 旋转数组
给定一个数组,将数组中的元素向右移动 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次。
但是这种解法效率太慢了,会超出时间限制。
代码示例:
void rotate1(int* nums, int numsSize, int k) {
k %= numsSize; //只移动余数次。
int tmp = 0;
int i;
for (i = 0;i < k;i++)
{
tmp = nums[numsSize - 1];
int j;
for (j = numsSize - 1;j > 0;j--)
{
nums[j] = nums[j - 1];
}
nums[0] = tmp;
}
}
解法2:三步逆序
假设要移动k次,将数组分为arr1和arr2两个数组,arr2的大小为k。分别逆序。
arr1和arr2逆序完毕,最后把整个数组逆序。
旋转完毕。
时间复杂度:n*k
空间复杂度:O(1)
代码示例
//实现逆序
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) {
k %= numsSize; //只让余数移动。节省性能。
reverse(nums, 0, 0 + (numsSize - k) - 1); //arr1旋转一次
reverse(nums, (numsSize - k), numsSize - 1);//arr2旋转一次
reverse(nums, 0, numsSize - 1);//整体旋转一次
}