旋转数组
力扣原题
方案一
循环K次将数组所有元素向后移动⼀位(代码不通过)
时间复杂度O(n2)
空间复杂度O(1)
void rotate(int* nums, int numsSize, int k) {while (k--) {int end = nums[numsSize - 1];for (int i = numsSize - 1; i > 0; i--) {nums[i] = nums[i - 1];}nums[0] = end;}
}
- 时间复杂度过高,需要优化
方案二
申请新数组空间,先将后k个数据放到新数组中,再将剩下的数据挪到新数组中
时间复杂度O(n)
空间复杂度O(n)
void rotate(int* nums, int numsSize, int k) {int newArr[numsSize];for (int i = 0; i < numsSize; ++i) {newArr[(i + k) % numsSize] = nums[i];}for (int i = 0; i < numsSize; ++i) {nums[i] = newArr[i];}
}
- 记得最后要把新数组数据复制到原数组
- 时间复杂度降低了,可以通过
- 空间复杂度提升,仍可以优化
方案三
数组翻转
该方法基于如下的事实:当我们将数组的元素向右移动 k 次后,尾部 kmodn 个元素会移动至数组头部,其余元素向后移动 kmodn 个位置。
我们可以先将所有元素翻转,这样尾部的 kmodn 个元素就被移至数组头部,然后我们再翻转 [0,kmodn−1] 区间的元素和 [kmodn,n−1] 区间的元素即能得到最后的答案
时间复杂度:O(n)
空间复杂度:O(1)
void reverse(int* nums, int begin, int end) {while (begin < end) {int tmp = nums[begin];nums[begin] = nums[end];nums[end] = tmp;begin++;end--;}
}
void rotate(int* nums, int numsSize, int k) {k = k % numsSize;reverse(nums, 0, numsSize - k - 1);reverse(nums, numsSize - k, numsSize - 1);reverse(nums, 0, numsSize - 1);
}