问题:旋转数组
给定一个数组,将数组中的元素向右移动 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();
}
}
};
运行结果截图:
执行耗时: