题目
题目来自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)
的原地算法。
实现
方法1
一个比较典型的旋转问题
以1234567
,k=3
举例:
先全部旋转reverse(0, length-1)
,结果:
7654321
然后旋转前k
个数字,reverse(0, k-1)
,结果:
567 4321
最后旋转剩余数字:,reverse(k, length-1)
,结果:
567 1234
用代码实现:
var rotate = function (nums, k) {
let length = nums.length;
// 处理k<n的情况
k = k % length;
// 旋转
function reverse(arr, start, end) {
while (start < end) {
[arr[start], arr[end]] = [arr[end], arr[start]];
start++;
end--;
}
}
reverse(nums, 0, length - 1);
reverse(nums, 0, k - 1);
reverse(nums, k, length - 1);
return nums
};
要注意k = k % length
这行代码,有两个作用,一个是防止数组的长度小于k
时出现错误,另一个是减少移动的次数
方法2
可以直接利用JS内置的插入的方法,它会自动将数组的序号更新(我怎么把这个方法忘了)
var rotate = function (nums, k) {
for (let i = 0; i < k; i++) {
nums.unshift(nums.splice(nums.length-1, 1)[0])
}
return nums
};
方法3
又想了一个特笨的方法
还是以1234567
,k=3
举例
先从后向前进行第一轮移动,每相邻两项进行换位,结果是:
7123456
然后再进行第二轮移动,类似的,结果是:
6712345
k
决定了移动的次数:
var rotate = function (nums, k) {
while (k) {
for (let i = nums.length - 1; i > 0; i--) {
[nums[i], nums[i-1]] = [nums[i-1], nums[i]]
}
k--;
}
return nums
};
应该还有其他的方法,比如通过递归实现,回头重新的时候再想吧。