初级算法探索——数组篇(三)

问题:旋转数组

给定一个数组,将数组中的元素向右移动 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) 的原地算法。

js解决办法:

testCallFunc3: function() {
    var rotateTable = [1,2,3,4,5,6,7];
    // var rotateTable = [-1,-100,3,99]
    this.moveRightTableBySteps3(rotateTable, 3);
    cc.log("rotated table:" + JSON.stringify(rotateTable));
},

// 思路一:将尾元素插入数组头,然后移除尾元素,重复steps遍操作
moveRightTableBySteps: function(rotateTable, steps) {
    steps = steps%rotateTable.length;
    if (rotateTable.length == 0 || steps == 0) {
        return;
    }
    for (var i = 0; i < steps; i++) {
        rotateTable.unshift(rotateTable[rotateTable.length-1]);
        rotateTable.pop();
    }
},

// 思路二:将头元素插入数组尾,然后移除头元素,重复length-steps遍操作
moveRightTableBySteps2: function(rotateTable, steps) {
    steps = steps%rotateTable.length;
    if (rotateTable.length == 0 || steps == 0) {
        return;
    }
    for (var i = 0; i < rotateTable.length-steps; i++) {
        rotateTable.push(rotateTable[0]);
        rotateTable.shift();
    }
},

// 思路三:向右移一位元素替换,重复steps遍操作
moveRightTableBySteps3: function(rotateTable, steps) {
    steps = steps%rotateTable.length;
    if (rotateTable.length == 0 || steps == 0) {
        return;
    }
    for (var i = 0; i < steps; i++) {
        var instead = rotateTable[rotateTable.length-1];
        for (var index = 0; index < rotateTable.length; index++) {
            var instead2 = rotateTable[index];
            rotateTable[index] = instead;
            instead = instead2;
        }
    }
},

将思路一和思路二进行融合优化,争取最少步数解决问题:

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {void} Do not return anything, modify nums in-place instead.
 */
var rotate = function(nums, k) {
    k = k%nums.length;
    if (nums.length == 0 || k == 0) {
        return;
    }
    if (k < nums.length/2) {
        for (var i = 0; i < k; i++) {
            nums.unshift(nums[nums.length-1]);
            nums.pop();
        }
    } else {
        for (var i = 0; i < nums.length-k; i++) {
            nums.push(nums[0]);
            nums.shift();
        }
    }
};

运行结果截图:
这里写图片描述
执行耗时:
这里写图片描述

猜你喜欢

转载自blog.csdn.net/zhenyu5211314/article/details/81067399