谈谈C++中vector的删除
在vector中有三种涉及删除的方法,第一种是vector::clear(),第二种是vector::erase(),第三种是vector::pop_back()。clear用来清空整个vector,同时将size变成0,无返回值;erase通过传入迭代器进行删除,既可以删除单个元素,也可以删除某一范围的元素,删除之后它将返回下一个位置的迭代器;pop_back用来删除末尾元素,同时将size减1,无返回值。
一般来讲erase用得比较多,不注意也容易踩坑,例如:
vector<int> vec{1, 4, 3, 2, 5, 2}; for(auto it = vec.begin(); it != vec.end(); it++) { if(*it < 3) { cout << "num is " << *it << endl; it = vec.erase(it); if(it == vec.end()) break; } }
这个代码如果不在for循环里加上最后的判断会导致错误,原因在于,当it指向最后一个2时,满足删除条件,删除后it变成了end,然后又做了++操作,导致越界。
类似的坑还有许多,这里就不一一列举了,总之,它们的共同点在于对vector进行删除时都不够优雅,需要进行一些额外的处理避免误操作。
事实上,我们完全可以使用<algorithm>提供的remove和remove_if方法,配合erase进行删除。这两个方法并不是真的删除元素,而是以覆盖的方式将满足删除条件的元素移到容器最后(它们具体的实现方式很有意思,但是这里从略,总之有两点:1)它们并没有真的删除元素;2)它们使用了“覆盖”的方式,将要删除的元素移到最后,因此这些元素可能因为被覆盖过而发生改变,但这已经不重要了),然后返回指向第一个要删除位置的迭代器。这样一来,删除就优雅多了,例如:
vec.erase(remove(vec.begin(), vec.end(), 3), vec.end());
如果再用上lambda,看起来会更酷:
vec.erase(remove_if(vec.begin(), vec.end(), [](int x){return x == 3;}), vec.end());
原文链接:
参考:
C++中Vector.erase()删除某一元素的正确用法
https://en.cppreference.com/w/cpp/algorithm/remove