Как перебрать набор STL и выборочно удалить элементы?

Question

Следующий код не работает правильно. Как это сделать правильно?

for (std::set<Color>::iterator i = myColorContainer.begin();
            i!=myColorContainer.end();
            ++i)
{
    if ( *i  == Yellow)
    {
        DoSomeProccessing( *i );
        myColorContainer.erase(i);
    }
}

Answer 1

Попробуйте:

for(std::set<Color>::iterator it = myColorContainer.begin(); 
    it != myColorContainer.end();) { // note missing it++
     if( (*it) == Yellow ) {
        DoSomeProcessing(*it);
        myColorContainer.erase(it++); // post increment (original sent to erase)
     }
     else {
       ++it; // more efficient than it++;
     }
}

Answer 2

Вам не нужен цикл, поскольку вы имеете дело с сетом.

std::set<Color>::iterator it = myColorContainer.find(Yellow);
if (it != it.myColorContainer.end()){
  DoSomeProcessing(*it);
  myColorContainer.erase(it);
}

Answer 3

for (std::set<Color>::iterator i = myColorContainer.begin();
            i!=myColorContainer.end(); /* No i++ */)
{
    if ( *i  == Yellow)
    {
        DoSomeProccessing( *i );
        std::set<Color>::iterator tmp = i;
        ++i;
        myColorContainer.erase(tmp);
    }
    else {
        ++i;
    }
}

Как только вы переходите к следующему сообщению с ++i, оно гарантированно будет действительным - свойство std::set, что итераторы на вставленных элементах никогда не становятся недействительными, если элемент не удален.

Так что теперьВы можете безопасно стереть предыдущую запись.