向量迭代器不可递增，这与迭代器和擦除的操作有关

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

删除最后一个元素后，it成为nums.end()，并增加这是不允许的。

擦除后不应该增加it。这也将解决擦除后一个元素被跳过的问题。

另外&在auto &it = nums.begin()应该被删除，因为它会使它非法初始化非const引用。

试试这个：

#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; 
}