c++ - vector iterator not incrementable，这与iterator和erase的操作有关

Question

我在网上找了很久。但是没有用。请帮助或尝试提供一些解决此问题的想法。

class Solution{
public:
    int removeElement(vector<int> &nums, int val)
    {

        for (auto &it = nums.begin(); it != nums.end(); ++it)
        {
            if (*it == val)
            {
                it = nums.erase(it);
            }
        }
        return nums.size();
    }
};
int main(void)
{
    Solution s; 
    vector<int> vi = { 3, 2, 2, 3 };
    cout << "size = " << s.removeElement(vi, 3) << endl;
    for (auto &i : vi)
    {
        cout << i << " ";
    }
    cout << endl;

    return 0;
}

这是我的代码的类体和主要函数体。但是当我运行它时，编译器弹出一个窗口:

Answer 1

删除最后一个元素后，它变成了nums.end()并且不允许增加它。

你不应该在删除后增加它。这也将解决删除后跳过一个元素的问题。

还应删除 auto &it = nums.begin() 中的 &，因为使用它会使非常量引用的初始化无效。

试试这个:

#include <iostream>
#include <vector>

using std::vector;
using std::cout;
using std::endl;

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();
    }
};
int main(void)
{
    Solution s; 
    vector<int> vi = { 3, 2, 3, 3, 2, 3 }; // added testcase of consecutive 3
    cout << "size = " << s.removeElement(vi, 3) << endl;
    for (auto &i : vi)
    {
        cout << i << " ";
    }
    cout << endl;

    return 0;
}