189. 旋转数组(JS实现)
189. 旋转数组(JS实现)
1 题目
给定一个数组,将数组中的元素向右移动 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) 的 原地 算法。
链接:https://leetcode-cn.com/problems/rotate-array
2 思路
这道题将数组元素向右移动k个位置,就是每个元素移动后的索引为i+k,因此,我们可以将每个元素都向右移动k个位置,即得到答案
3代码
/**
* @param {number[]} nums
* @param {number} k
* @return {void} Do not return anything, modify nums in-place instead.
*/
var rotate = function(nums, k) {
let len = nums.length;
k = k % len;
if (k === 0) return;
let count = 0;
for (let start=0; count < len; start++) {
let current = start;
let next = start + k;
let prev = nums[current]; //记录当前元素值
do {
let temp = nums[next]; //记录下个元素值
nums[next] = prev; //替换元素
prev = temp;
current = next;
next += k;
next %= len; //索引超出时回到数组开头
count++; //记录已经移动的元素的个数
} while(current !== start)
}
};