本文已参与「新人创作礼」活动,一起开启掘金创作之路。
题目
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
输入:nums = [3,2,2,3], val = 3
输出:2, nums = [2,2]
解释:函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。例如,函数返回的新长度为 2 ,而 nums = [2,2,3,3] 或 nums = [2,2,0,0],也会被视作正确答案。
题目解析
思路一
我们可以利用双指针,先将 nums 中等于 val 的元素移动到数组末尾,当 left=right 时, 如果 nums[left]等于 val, 即删除 nums[left], 返回 left 的值即可;如果 nums[left]不等于 val, 即不删除 nums[left], 返回 left+1
var removeElement = function (nums, val) {
if (!nums.length) return 0;
let left = 0,right = nums.length - 1;
while (left < right) {
// 比较val和左指针指向的元素
while (left < right && nums[left] !== val) {
left++;
}
// 比较val和右指针指向的元素
while (left < right && nums[right] === val) {
right--;
}
// 跳出内部循环时, nums[left] === val、nums[right] !== val
// 交换nums[left]和nums[right],此时数组尾部全是值为val的元素
[nums[left], nums[right]] = [nums[right], nums[left]];
}
// 判断left===right时,nums[left]和val的关系
return nums[left] === val ? left : left + 1;
};
思路二
循环nums数组,如果当前值不等于val,则nums[m++]进行赋值当前值,顺便m+1进行下一次对比时赋值,最后返回m(不需要加1,因为m++的++在后面,结束循环时会比循环时大1)
var removeElement = function (nums, val) {
const len = nums.length;
if (len === 0) return 0;
let m = 0;
for (let i = 0; i < len; i++) {
if (nums[i] !== val) {
nums[m++] = nums[i];
}
}
return m;
};