STL中容器元素的删除

2015-03-31  本文已影响764人  AlvinL

STL中的容器可以分为两类:一类是按照数组方式连续存储元素的容器,如vectordeque,另一类是按照不连续的节点存储元素的容器,如setmaplist等。
那么这两类容器删除元素的时候需要注意一个问题,使用iterator指定删除某个元素的时候,要注意返回的iterator的值。下面分别举setvector的例子来说明。

set删除元素

假如要删除set中满足某个条件的元素,这时候需要遍历,看代码:

set<int> s;
s.insert(1);
s.insert(4);
s.insert(2);
s.insert(3);

for (set<int>::iterator it = s.begin(); it != s.end(); it ++)
    cout << *it << endl;

for (set<int>::iterator it = s.begin(); it != s.end(); )
{
    if (*it % 2 != 0) // 删除奇数
        s.erase(it++);
    else
        it ++;
}

大家注意到s.erase(it++)这一句,it++返回是+1之前的it的值,那么此时删除的正是满足条件的元素,而it本身又进行了+1,指向了下一个元素,那你可能会问为什么不直接删除之后再+1呢,如下代码:

s.erase(it);
it++;

这样为什么不行呢?因为在erase之前,it其实指向的还是原来的元素,只不过它已经不存在set中,那么此时it++,得到的值不是set中的那一个元素,而是不确定的,对这个指针进行访问,会得到意想不到的结果,甚至导致程序崩溃。

vector删除元素

vector内部存储元素是采用数组方式存储的,即是连接的,删除元素可以按照以下方法:

vector<int> ve;
ve.push_back(1);
ve.push_back(3);
ve.push_back(2);
ve.push_back(4);

for (vector<int>::iterator it = ve.begin(); it != ve.end();)
{
    if (*it % 2 != 0)
        ve.erase(it);
    else
        it ++;
}

vectorset不一样的地方在于,调用erase之后,it自动指向下一个元素,所以在遍历删除的时候需要注意,erase之后不需要再进行it++,否则会跳过了一个元素。

上一篇 下一篇

猜你喜欢

热点阅读