遍历向量并删除某些元素的正确方法是什么[重复]

Question

我遇到了一个问题，我想遍历我的向量并删除不再需要的元素。它失败的原因很明显，但是当我尝试我的幼稚方法时并没有看到它。基本上，当我删除一个元素时，迭代器会失效，并且循环无法继续。我所做的如下：

    #define GOOD 1
    #define BAD 0

    struct Element
    {
        Element(int isGood) : good(isGood){}
        bool good;
    };

    int main()
    {
        std::vector<Element> arr;
        arr.push_back(Element(BAD));
        arr.push_back(Element(GOOD));
        arr.push_back(Element(BAD));
        arr.push_back(Element(GOOD));

    //__CLEAN ARRAY__//
        for (auto it = arr.begin(); it != arr.end(); ++it)
        {
            if ((*it).good == false) arr.erase(it);
        }
    }

所以很明显这不起作用，我想知道这样做的正确/最佳方法是什么。如果没有找到好的，我的下一步是用新的迭代器重新启动循环，但这似乎也是一种浪费。理想情况下，循环会在它停止的地方继续使用新的迭代器？

谢谢。

Answer 1

你想要：

arr.erase( std::remove_if(arr.begin(), arr.end(), [](auto& obj){return obj.good == false;}), arr.end() );

及其所谓的remove-erase idiom：

https://en.wikipedia.org/wiki/Erase%E2%80%93remove_idiom

但是如果你想修复循环那么它是可能的，erase 返回下一个有效的迭代器，所以你应该使用它：

    for (auto it = arr.begin(); it != arr.end(); )
    {
      if ((*it).good == false) 
        it = arr.erase(it);
      else
        it++;
    }