给定一个数组
nums,编写一个函数将所有0移动到数组的末尾,同时保持非零元素的相对顺序。请注意 ,必须在不复制数组的情况下原地对数组进行操作。
示例 1:
输入: nums = [0,1,0,3,12] 输出: [1,3,12,0,0]示例 2:
输入: nums = [0] 输出: [0]
题解1 删除再添加
思路
可以先统计出0的个数,并在统计时删除0。最后在末位补0即可。
要注意如果从头遍历,使用 splice 删除时会打乱遍历顺序,所以要从后遍历。
代码
function moveZeroes(nums: number[]): void {
let count: number = 0;
for (let i = nums.length - 1; i >= 0; i--) {
if (nums[i] === 0) {
nums.splice(i, 1);
count++;
}
}
for (let i = 0; i < count; i++) {
nums.push(0);
}
};
时空复杂度分析
时间复杂度:一次遍历加上答案常数,所以为 O(n)
空间复杂度:只使用了1个变量,所以是 O(1)
题解2 双指针
思路
这题的难点在于原地对数组进行操作,所以上面的增删数组是不允许的。
既然是原地,那么肯定是使用 swap 来交换两个数的位置。而首先我们要找到这两个数,于是需要有两个指针,所以这题可以用双指针来做。
指针 i 和指针 j 从头遍历,只要 i 不是 0 就进行交换,并且 j 指针往前走。
代码
function moveZeroes(nums: number[]): void {
let j: number = 0;
for (let i = 0; i < nums.length; i++) {
if (nums[i] !== 0) {
const temp: number = nums[j];
nums[j++] = nums[i];
nums[i] = temp;
}
}
};
时空复杂度分析
时间复杂度:一次遍历 O(n)
空间复杂度:O(1)