C++ STL常用容器删除操作注意事項

C/C++程序员对STL 容器肯定不会陌生, 以下是对常用容器进行 erase 操作注意事项的总结:

vector

vector 容器是用数组实现的, 它在内存是连续分布的.对它进行 insert 或 erase 操作涉及到对插入点或删除点后的元素进行移位, 所以插入点或删除点后的iterator 会失效.

iterator erase( iterator pos );iterator erase( const_iterator pos );

查看 STL 帮助得知vector::erase 函数的返回值是当前迭代器指针的下一个迭代器.

Iterator following the last removed element. If the iterator pos
refers to the last element, the end() iterator is returned.

所以对 vector 进行 erase 操作可以采用如下形式:

//删除 vector1中值等于 num 的元素for(; it != vector1.end(); ) {    if(num == *it)        it = vector1.erase(it);    else        ++it;}

list

STL 的list容器是采用双向链表实现的, 使用不连续的内存空间存储元素.inser时不会导致任何迭代器失效, erase 时删除点的迭代器失效, 但是不影响其他迭代器. 它的 erase 函数类似 vector, 该函数返回值是当前迭代器指针的下一个迭代器.但是由于底层实现不同, list 的删除操作可以采用更加灵活地方式.

//删除 list1中值等于 num 的元素for(; it != list1.end(); ) {    if(num == *it)        it = list1.erase(it);        //list1.erase(it++); //also OK    else        ++it;}

map

STL的 map 容器底层采用红黑树实现, 也就是说是一种平衡二叉树, 它的插入和删除效果和 list 类似.inser时不会导致任何迭代器失效, erase 时只有删除点的迭代器失效, 其他迭代器不受影响. 代码写法也与 list 类似:

 // 删除 map1中所有 key 为奇数的元素for(auto it = map1.begin(); it != map1.end(); ) {            if(it->first % 2 == 1)        //it = map1.erase(it);        map1.erase(it++);    else        ++it;}

其他容器如 dequeue的操作方法借鉴vector, set 可以借鉴 map 这里就不再赘述.一下是测试代码:

//To complie: g++ -o test test.cpp -std=c++11#include <iostream>#include <algorithm>#include <string>#include <list>#include <vector>#include <map>using std::cout;void test1() {    std::vector<int> vi{1,2,3,3,3,5};    int num = 3;    std::vector<int>::iterator it = vi.begin();    //erase element whose value is 3    for(; it != vi.end(); ) {        if(num == *it)            it = vi.erase(it);        else            ++it;    }    std::copy(vi.begin(), vi.end(), std::ostream_iterator<int>(cout, "\t"));    cout << '\n';}void test2(){    std::list<int> li{1,2,3,3,3,5};    int num = 3;    std::list<int>::iterator it = li.begin();    //erase element whose value is 3    for(; it != li.end(); ) {        if(num == *it)            //it = li.erase(it);            li.erase(it++);        else            ++it;    }    std::copy(li.begin(), li.end(), std::ostream_iterator<int>(cout, "\t"));    cout << '\n';}void test3(){    std::map<int, std::string> map1 = {{1, "one"}, {2, "two"}, {3, "three"},                                    {4, "four"}, {5, "five"}, {6, "six"}};    // erase all odd numbers from map1    for(auto it = map1.begin(); it != map1.end(); ) {                if(it->first % 2 == 1)            //it = map1.erase(it);            map1.erase(it++);        else            ++it;    }    for(auto& p : map1) {        cout << p.second << ' ';    }           cout << '\n';}int main() {    // test1();    // test2();    test3();    return 0;}