Удалить единственный элемент вектора во время итерации? - PullRequest
Новая
       15

Удалить единственный элемент вектора во время итерации?

1 голос
/ 09 мая 2019

Я немного запутался в итераторах. Я хотел бы удалить член вектора, который имеет определенное значение, даже если это единственный элемент в векторе.

У меня есть вектор класса. Я хотел бы, чтобы одна из переменных члена класса, например width, имела определенное значение, этот экземпляр класса будет окончательно удален из вектора. Давайте предположим, что класс состоит только из конструктора, принимая width и height, как открытые переменные, так и деструктор в качестве своих открытых функций. 

vector<Rectangle> vect1;

Rectangle memVar1(3, 5);
Rectangle memVar2(6, 10);

vect1.push_back(memVar1);
vect1.push_back(memVar2);

Таким образом, он будет перебирать все элементы вектора, ища, если высота равна 5. 

for(std::vector<myClass>::iterator it = vect1.begin(); it != vect1.end(); ++it) {
    if (it->height == 5) {
    //delete the element from the vector, and so memVar2 will be the only element and it will be in slot 0 of the vector now.
    it = vect1.erase(it);
    } 
}

И это работает, как и ожидалось, проблема возникает, если в векторе есть только один элемент, который имеет нежелательную высоту 5. Я хотел бы удалить элемент из вектора, но затем также безопасно выйти из итерации. , В приведенном выше примере я мог бы вставить break после установки его в vect1.erase (it), но это не сработало бы, если бы мое стирание было, например, внутри переключателя внутри цикла итерации, и поэтому мне понадобилось бы условная переменная для выхода оттуда, когда она покидает оператор switch.

Так что мне было интересно, есть ли способ сделать это, который не требует от меня использования break, чтобы избежать появления кода беспорядка, если я стираю внутри оператора switch?

1 Ответ

4 голосов
/ 09 мая 2019

При удалении элементов в цикле необходимо учитывать, что при удалении элемента с индексом i следующий элемент находится с индексом i, а не с индексом i+1.Кроме того, удаление элемента в векторе делает недействительными все итераторы в и после позиции, которую вы стерли (фактически, вы уже это учитываете).Следовательно, цикл должен выглядеть примерно так: 

for(auto it = vect1.begin(); it != vect1.end(); /* no increment here */) {
    if (it->height == 5) {
        it = vect1.erase(it);
    } else {
        ++it;
    }
}

PS: только после написания ответа я понял, что мое объяснение поддельное.Правильное объяснение состоит в том, что erase уже возвращает итератор для элемента после того, который вы стерли, поэтому вы не должны увеличивать его.

...