题目描述

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

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

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

• 力扣题目链接

• 代码随想录链接

本题思路

暴力解法

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int size = nums.size();
        for (int i = 0; i < size; i++) {
            if (nums[i] == val) { // 发现需要移除的元素，就将数组集体向前移动一位
                for (int j = i + 1; j < size; j++) {
                    nums[j - 1] = nums[j];
                }
                i--; // 目前的nums[i]改变，原来的i后一位，保证下次循环会重新判断当前更改完的元素
                size--; // 此时数组的大小-1
            }
        }
        return size;

    }
};

注意细节

双指针法

慢指针：获取要更新的元素

快指针：获取数组元素

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int slow = 0, fast = 0;
        for (fast = 0; fast < nums.size(); fast++) {
            if (nums[fast] != val) {
                nums[slow] = nums[fast];
                slow++;
            }
        }
        return slow;
    }
};

如果快指针指到要移除的元素就不执行if判断，快慢指针就产生差距，下次循环如果不是要移除的元素即会覆盖慢指针所指的元素(第一理解)

快指针来找目标，如果非目标，利用慢指针来更新数组(第二理解)