移除数组中指定值的元素

来源：力扣

给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。

不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并 原地 修改输入数组。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

示例 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]，也会被视作正确答案。

---

这道题用Python来解决简直就是api调用大师

一开始我想的很简单，无非就是遍历数组，找到目标值就删除，不过此时存在一个问题，即删除和遍历不能同时进行， 删除的同时数组长度会变化，但是指针是递增的，所以存在连续目标值时候会出错，即删不干净，错误示例如下：

def removeElement(nums, val) -> int:
        for x in nums:
            if x == val:
                nums.remove(x)
        return len(nums)

想到这个问题后，可以通过列表的count方法计算目标值出现的次数，随后调用这么多次remove方法即可，毫无技术含量：

def removeElement(nums, val) -> int:
        count = nums.count(val)
        i = 0
        while i < count:
            nums.remove(val)
            i += 1
        return len(nums)

实际上，应该使用双指针来解决，右指针指向当前将要处理的元素，左指针指向下一个将要赋值的位置。

• 如果右指针指向的元素不等于 目标值，它一定是输出数组的一个元素，我们就将右指针指向的元素复制到左指针位置，然后将左右指针同时右移；
• 如果右指针指向的元素等于 目标值，它不能在输出数组里，此时左指针不动，右指针右移一位。

class Solution {
    public int removeElement(int[] nums, int val) {
        int n = nums.length;
        int left = 0;
        for (int right = 0; right < n; right++) {
            if (nums[right] != val) {
                nums[left] = nums[right];
                left++;
            }
        }
        return left;
    }
}

还有一种方法可以让两个指针从两头向中间遍历

如果左指针 left 指向的元素等于val，此时将右指针 right 指向的元素复制到左指针 left 的位置，然后右指针 right 左移一位。如果赋值过来的元素恰好也等于 val，可以继续把右指针 right 指向的元素的值赋值过来（左指针 left 指向的等于 val 的元素的位置继续被覆盖），直到左指针指向的元素的值不等于val 为止。

当左指针 left 和右指针 right 重合的时候，左右指针遍历完数组中所有的元素。

这样的方法两个指针在最坏的情况下合起来只遍历了数组一次。与方法一不同的是，方法二避免了需要保留的元素的重复赋值操作。

class Solution {
    public int removeElement(int[] nums, int val) {
        int left = 0;
        int right = nums.length;
        while (left < right) {
            if (nums[left] == val) {
                nums[left] = nums[right - 1];
                right--;
            } else {
                left++;
            }
        }
        return left;
    }
}