# [189] 轮转数组
给你一个数组,将数组中的元素向右轮转 k 个位置,其中 k 是非负数。
示例 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]
示例 2:
输入:nums = [-1,-100,3,99], k = 2
输出:[3,99,-1,-100]
解释:
向右轮转 1 步: [99,-1,-100,3]
向右轮转 2 步: [3,99,-1,-100]
提示:
1 <= nums.length <= 10^5
-2^31 <= nums[i] <= 2^31 - 1
0 <= k <= 10^5
进阶:
尽可能想出更多的解决方案,至少有 三种 不同的方法可以解决这个问题。
你可以使用空间复杂度为 O(1) 的 原地 算法解决这个问题吗?
这道题要求使用原地算法,因此我们必须注意处理数据的先后顺序。因此,我们需要首先将被原地替换的数据优先存储起来,防止被覆盖。
最直观的想法是,将轮转后从尾部替换到头部的数据都记录下来,存成一个数组tempArr。之后,将这些位置上的数据替换成正确的数据。最后,将记录下来的数据从头开始,依次覆盖nums中的对应位置。
function rotate(nums: number[], k: number): void {
const len = nums.length;
k = k % len;
const tempArr = [];
for (let i = len - k; i < len; i++) {
tempArr.push(nums[i]);
}
for (let i = len - 1; i >= k; i--) {
nums[i] = nums[i - k];
}
for (let i = 0; i < tempArr.length; i++) {
nums[i] = tempArr[i];
}
}
这样的时间复杂度为O(3k),应该与要求的时间复杂度O(n)在同一水平上。但是这里需要O(k)的空间复杂度。如果要求使用O(1)的空间,就得另想办法了。
我们注意到,当我们将nums中的元素向右移动 k 次后,尾部的k个元素会移动至头部,其余元素向后移动 k 个位置。
而将nums翻转后,尾部的k个元素恰好就放在了前k个位置。而此时将前k个元素,与后面的元素分别再做依次翻转,即可得到最后的结果。
此时,问题就变成了普通的数组翻转问题了,使用双指针就能搞定。
此时的空间复杂度为O(1),时间复杂度为O(2n)。
function rotate(nums: number[], k: number): void {
const len = nums.length;
k = k % len;
reverse(0, len - 1);
reverse(0, k - 1);
reverse(k, len - 1);
function reverse(start: number, end: number) {
let left = start;
let right = end;
while (left < right) {
const temp = nums[left];
nums[left] = nums[right];
nums[right] = temp;
left++;
right--;
}
}
}