在for循环中使用std::vector::erase()删除元素,需要特别注意迭代器的变化。
例如,这个代码看似正常,但实际上有问题:
std::vector<int> vec = {1, 2, 3, 4};
for (int i = 0; i < vec.size(); i++) {
if (vec[i] == 2) {
vec.erase(vec.begin() + i);
}
}
在上面的代码中,erase()被调用后,vec的大小减小了1,但i并没有相应减小。这会导致下次访问vec[i]时索引越界,产生未定义行为。
正确的方法是使用迭代器,并在erase后更新迭代器:while方法
std::vector<int> vec = {1, 2, 3, 4};
auto it = vec.begin();
while (it != vec.end()) {
if (*it == 2) {
it = vec.erase(it);
} else {
it++;
}
}
这里,我们使用迭代器it在vec中遍历。一旦调用erase删除一个元素,erase会返回下一个元素的迭代器,我们使用它来更新it。
这样,it会随着vec大小的变化而变化,避免了访问无效索引的问题。
1、for方法
std::vector<int> vec = {1, 2, 3, 4};
for (auto it = vec.begin(); it != vec.end(); ) {
if (*it == 2) {
it = vec.erase(it);
} else {
++it;
}
}
2、for方法
std::vector<int> vec = {1, 2, 3, 4};
for (auto it = vec.begin(); it != vec.end(); ++it) {
if (*it == 2) {
it = vec.erase(it);
--it;
}
}