LeetCode 27. 移除元素（JavaScript 简单）

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

/**
 * @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。

/**
 * @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。

文章来源地址https://uudwc.com/A/3wBNO