Мне нужно пройти через набор и удалить элементы, которые соответствуют предопределенным критериям.
Это тестовый код, который я написал:
#include <set>
#include <algorithm>
void printElement(int value) {
std::cout << value << " ";
}
int main() {
int initNum[] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 };
std::set<int> numbers(initNum, initNum + 10);
// print '0 1 2 3 4 5 6 7 8 9'
std::for_each(numbers.begin(), numbers.end(), printElement);
std::set<int>::iterator it = numbers.begin();
// iterate through the set and erase all even numbers
for (; it != numbers.end(); ++it) {
int n = *it;
if (n % 2 == 0) {
// wouldn't invalidate the iterator?
numbers.erase(it);
}
}
// print '1 3 5 7 9'
std::for_each(numbers.begin(), numbers.end(), printElement);
return 0;
}
Сначала я думал, что стирание элемента из набора во время итерации через него приведет к недействительности итератора, а приращение в цикле for будет иметь поведение undefined. Несмотря на это, я выполнил этот тестовый код, и все прошло хорошо, и я не могу объяснить, почему.
Мой вопрос: Является ли это определенным поведением для std-наборов или эта реализация специфична? Кстати, я использую gcc 4.3.3 на ubuntu 10.04 (32-разрядная версия).
Спасибо!
Предлагаемое решение:
Является ли это правильным способом для итерации и стирания элементов из набора?
while(it != numbers.end()) {
int n = *it;
if (n % 2 == 0) {
// post-increment operator returns a copy, then increment
numbers.erase(it++);
} else {
// pre-increment operator increments, then return
++it;
}
}
Изменить: ПРЕДПОЧТИТЕЛЬНОЕ РЕШЕНИЕ
Я придумал решение, которое кажется мне более изящным, хотя он делает то же самое.
while(it != numbers.end()) {
// copy the current iterator then increment it
std::set<int>::iterator current = it++;
int n = *current;
if (n % 2 == 0) {
// don't invalidate iterator it, because it is already
// pointing to the next element
numbers.erase(current);
}
}
Если в то время есть несколько условий тестирования, каждый из них должен увеличивать итератор. Мне нравится этот код лучше, потому что итератор увеличивается только в одном месте, делая код менее подверженным ошибкам и более удобочитаемым.
