STL vector使用erase注意点

在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;
    } 
}