Каков правильный способ итерации вектора и удаления определенных элементов

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()); 

и его называют удалить стирания идиома:

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

Но если вы хотите, чтобы исправить цикл, то это возможно, удалить возвращает следующий действительный итератора, чтобы вы должен использовать его:

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