给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
示例:
输入: [0,1,0,3,12]
输出: [1,3,12,0,0]
说明:
- 必须在原数组上操作,不能拷贝额外的数组。
- 尽量减少操作次数。
解法
题目说道在原数组上操作,不能出现新的数组。最开始我想到的是遍历,并在遍历的过程中碰到 0 就 splice(i, 1) 去掉后在 push。这么操作的确挺容易想到,但 splice 的时间复杂度O(n),毕竟是要移动后面所有元素的位置的。
优化的思想就是采用快慢指针,图片出自动画演示 283.移动零,时间复杂度 O(n) 并且空间复杂度 O(1),的确厉害。
代码
/**
* @param {number[]} nums
* @return {void} Do not return anything, modify nums in-place instead.
*/
var moveZeroes = function(nums) {
let index = 0;
for (let i = 0, len = nums.length; i < len; i++) {
if (nums[i] === 0) {
index += 1;
} else if (index !== 0) {
nums[i - index] = nums[i];
nums[i] = 0;
}
}
};