移除元素|TypeScript 代码笔记
题目还原
自己回忆:
给定一个nums数组(可能有重复元素)和val。要求删除nums中值为val的元素,并返回新数组的长度。
LeetCode:
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于val的元素。元素的顺序可能发生改变。然后返回 nums 中与 val 不同的元素的数量。
假设 nums 中不等于 val 的元素数量为 k,要通过此题,您需要执行以下操作:
- 更改
nums数组,使nums的前k个元素包含不等于val的元素。nums的其余元素和nums的大小并不重要。 - 返回
k。
思路分析
给出两种方法:暴力解法和双指针法。
第一种:暴力解法
大体思路
- 第一层for循环:定位到要删除的目标元素。
- 第二层for循环:从目标元素的下一个元素开始进行向前的覆盖操作。
注意点
- 每定位到一次目标元素时,就要更新数组的长度。第一层for循环和第二层for循环的边界都依赖最新数组的长度。
- 每定位到一次目标元素时,第一层for循环的迭代表达式会使得
i++。这将会漏掉一个元素。所以需要在每定位到一次目标元素时,主动进行i--操作。
代码解答
const removeElement1 = (nums: number[], val: number): number => {
let len = nums.length; // 记录最新数组长度
for (let i = 0; i < len; i++) {
if (nums[i] === val) {
// 向前的覆盖操作
for (let j = i + 1; j < len; j++) {
nums[j - 1] = nums[j];
}
len--; // 更新最新数组长度
i--; // 防止漏掉一个元素没有遍历
}
}
return len;
};
第二种:双指针法
大体思路
给出一个定义,新数组:删除完所有值为val的元素后的最终数组。
- 快指针:指向新数组中的元素。
- 慢指针:指向新数组中的元素要去的位置。
优先遍历快指针,所以for循环的迭代对象是快指针。因为旧数组中的每一个元素都有可能成为新数组中的元素,所以快指针一定要遍历旧数组中的每一个元素。因此for循环的边界是固定的nums.length,区别于暴力解法中的两个for循环的边界是依赖最新数组的长度。
注意点
最初慢指针是0,每次遇到新数组中的元素时,慢指针会加一,当最后一次遇到新数组中的元素时,慢指针又会加一;所以慢指针最终会指在新数组最后一个元素的屁股后面。我们知道数组元素的角标是从0到length - 1,所以屁股后面对应的角标刚好是length。所以慢指针最终就是新数组的length。
代码解答
const removeElement2 = (nums: number[], val: number): number => {
let slow = 0, fast = 0;
for (; fast < nums.length; fast++) {
if (nums[fast] !== val) {
nums[slow] = nums[fast];
slow++;
}
}
return slow;
}
两个代码均运行成功✅,运行结果: