力扣每日一题-1-27.移除元素

今天是2022.5.16，你刷题了吗？

---

题目：

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

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

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

分析：

给定一个数组，我们需要删除数组里面指定元素，删除之后的数组，大小变为减去指定元素的大小，数组内部元素值顺序无所谓。

大小：原大小-指定数个数

顺序：不要求

解题：

我打算通过迭代器遍历该数组，找到对应的值，然后删除该迭代器指向的元素。

遍历：auto  迭代器定义

删除：erase

推荐写方法一，毕竟 auto写起来更方便

//方法一
class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        for (auto it = nums.begin(); it != nums.end();) 
        {
            if (*it == val) 
            {
                it = nums.erase(it);
            }
            else 
            {
                it++;
            }
        }
        return nums.size();
    }
};

//方法二
class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        for (vector<int>::iterator it = nums.begin(); it != nums.end();)
        {
            if (*it == val)
            {
                it= nums.erase(it);//返回值是当前被删除元素的下一个元素的迭代器
            }
            else {
                it++;
            }
        }
        return nums.size();
    }
};

问题：

有一个问题：删除 it 之后，it 已经无效，此时执行 it++就会报错，因此循环遍历的时候不能写 it++

解决：

因为erase之后，返回的是被删除元素的下一个元素的迭代器，因此，写为：

it= nums.erase(it);//返回值是当前被删除元素的下一个元素的迭代器

此时的it已经变为删除元素的下一个it了，此时也不需要在 for循环 执行 it++