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)
    }
};

猜你喜欢

转载自blog.csdn.net/zjw_python/article/details/107879990
今日推荐