C++中STL容器利用迭代器删除元素小结

C++中STL容器利用迭代器删除元素小结

在STL容器中删除元素，容易导致迭代器失效，故应该清楚明白其用法，现在总结一下。
转载自：https://blog.csdn.net/yf_li123/article/details/75003425#comments

1.关联容器

对于关联容器(如map, set, multimap, multiset), 删除当前的iterator, 仅仅会使当前的iterator失效，只要在erase时，递增当前的iterator即可。这是因为map之类的容器，使用了红黑树来实现，插入，删除一个结点不会对其他结点造成影响。使用方式如下例子：
一个例子

#include<iostream>
#include<set>
using namespace std;

int main() {
    set<int> valset = { 1,2,3,4,5,6 };
    set<int>::iterator iter;
    for (iter = valset.begin(); iter != valset.end(); ) {
        if (*iter == 3)
            valset.erase(iter++);
        else if (*iter == 4)
            valset.erase(iter++);
        else if (*iter == 6)
            valset.erase(iter++);
        else
            ++iter;
    }
    for (auto i : valset) 
        cout << i << endl;
    return 0;
}

运行结果如下所示：

【注意】因为传给erase的是iter的一个副本，iter++是下一个有效的迭代器。

顺序容器

对于序列式容器（如vector，deque，list等），删除当前的iterator会使后面所有元素的iterator都失效。这是因为vector，deque使用了连续分配的内存，删除一个元素导致后面所有的元素会向前移动一个位置。不过erase方法可以返回下一个有效的iterator。使用方式如下，例如：

#include<iostream>
#include<vector>
using namespace std;

int main() {
    vector<int> val = { 1,2,3,4,5,6 };
    vector<int>::iterator iter;
    for (iter = val.begin(); iter != val.end(); ) {
        if (*iter == 3)
            iter = val.erase(iter);     //返回下一个有效的迭代器，无需+1
        else if (*iter == 4)
            iter = val.erase(iter);     //返回下一个有效的迭代器，无需+1
        else if (*iter == 6)
            iter = val.erase(iter);     //返回下一个有效的迭代器，无需+1
        else
            ++iter;
    }
    for (auto i : val)
        cout << i << endl;
    return 0;
}

运行结果如下所示：