1. 题目
给你一个数组
nums
和一个值val
,你需要 原地 移除所有数值等于val
的元素,并返回移除后数组的新长度。不要使用额外的数组空间,你必须仅使用
O(1)
额外空间并 原地修改输入数组。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
说明:
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以**「引用」**方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
你可以想象内部操作如下:
// nums 是以“引用”方式传递的。也就是说,不对实参作任何拷贝 int len = removeElement(nums, val); // 在函数里修改输入数组对于调用者是可见的。 // 根据你的函数返回的长度, 它会打印出数组中 该长度范围内 的所有元素。 for (int i = 0; i < len; i++) { print(nums[i]); }
示例 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],也会被视作正确答案。
示例 2:
输入:nums = [0,1,2,2,3,0,4,2], val = 2 输出:5, nums = [0,1,4,0,3] 解释:函数应该返回新的长度5,并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。
提示:
0 <= nums.length <= 100
0 <= nums[i] <= 50
0 <= val <= 100
2. 思路及代码
2.1 原思路和代码
利用双指针,一个指针p1指向要被替代的位置(此位置的值为 val),另一指针p2指向非val值nums[p2]。让 p2 指向的值替换掉 p1 位置上的原值。
注意点:
- p2 要保持在 p1 之后
- 由于 JavaScript 的数组不存在越界,要时常注意判断 p1 和 p2 是否越界了
- 另外最终返回数组的长度,所以要搞清楚返回的边界是 p1 还是 p1-1
javascript
/**
* @param {number[]} nums
* @param {number} val
* @return {number}
*/
var removeElement = function(nums, val) {
let p1=0, p2=0; //p1指向当前可能要被替换的位置,p2用遍历数组找到应该存下来的值
n = nums.length
while(p2<n && p1<n){
while(p1<n && nums[p1]!=val){
p1++; //找到当前值为 val 的位置 p1
}
for(p2=p1+1; nums[p2]==val; p2++); //找到当前值不为 val 的值 nums[p2]
if(p2>=n || p1>=n) break;
nums[p1] = nums[p2];
nums[p2] = val;
p1++;
p2++;
}
return p1;
};
2.2 反思和升级代码
- 判断+移动 是两次计算,如果直接 移动 可以解决的,额外再判断一次浪费时间。所以可以不需要判断 p2 指向的值是不是 val。如果 p2 指向的值是 val,直接复制过来,只要 当 p1 指向的值不是 val 时 p1 才动即可。
- 如果 p1 从数组起点开始,p2 从数组末尾开始,循环直到 p1===p2,这样总共循环只有 n(数组长度) 次。
- 将 p2 位置上的值复制到 p1 位置上时注意。如果保证 p2 当前指向的值已经被复制到 p1 上(先挪动再复制),则最终 p1 和 p2 重合时,++此时重合位置已经处理过++,直接返回的p1就是数组长度
- 如果是 p1 挪到 p2 上,p2指向的位置已被处理,则p1,p2此时指向的元素已经被复制到前面位置上,此时p1的下标就是数组长度;
- 如果是 p2 挪到 p1 上,此时p2指向的位置已经被处理,p1和p2指向的位置就是上一步自己复制给自己的操作,nums[left]===val,当前元素是val,返回的p1除了自己当前的位置之前的所有位置都不是val,返回的 p1 就是数组长度。
- p2 指向位置的值可能还没有被复制到 p1 上,当 p1 与 p2 重合时,还需要额外再讨论一次,当前 p1 指向的值是 val 时,返回的数组长度为 p1;当前 p1 指向的值不是 val 时,返回数组的长度应该是 p1+1。
双指针优化
left 指针(规定, 如果需要两个指针且存在 p1<p2 的循环结束条件时,命名为left 和right)从数组索引 0 开始遍历,right 指针从数组索引 num.length()-1 开始遍历。当 nums[left] === val 时,将 nums[right - 1] 的值复制过来,right--;当 nums[left] !== val 时,left++。重复上句操作,直到 left===right 时循环终止,返回 left。
javascript
/**
* @param {number[]} nums
* @param {number} val
* @return {number}
*/
var removeElement = function(nums, val) {
let left=0, right=nums.length; //left指向当前可能要被替换的位置,right用遍历数组找到应该存下来的值
while(left < right){
if(nums[left] == val) nums[left] = nums[--right];
else left++;
}
return left
};
3. 总结经验
- 能不判断解决问题的,不要判断。获得最好的时间复杂度,不需要过细考虑。
- 在块作用域内起作用的变量,使用 let。