码迷,mamicode.com
首页 > 其他好文 > 详细

vector删除操作 erase方法注意事项

时间:2015-03-07 18:46:55      阅读:683      评论:0      收藏:0      [点我收藏+]

标签:vector   stl   删除   erase   内存释放   

vector使用十分方便,可以存放任意类型数据结构。有时候我们会在vector中存放对象指针等,在删除某个元素的时候,我们还需要释放相应的内存。

本文主要谈一下erase方法,尤其是在循环体中使用erase方法,需要注意是否存在隐性问题。

erase的函数原型有两种形式:

iterator erase(iterator position);
iterator erase(iterator first, iterator last);

例如有一个类A,

class A
{
public:
    int id;
    A(void);
    ~A(void);
};

定义vector<A*> vec

for (int i=0; i<10; i++)
{
    A *p = new A();
    p->id = i;
    vec.push_back(p);
}

下面要删除vec中id为5的元素:

for(vector<A*>::iterator iter=vec.begin(); iter!=vec.end(); iter++)
{
    if(5 == (*iter)->id)
    {
        delete *iter;
        veci.erase(iter);
    }
}

初看这段代码没什么问题,实际上其中存在很大的问题;当调用erase方法后,iter就变成了一个野指针,继续循环iter++就出错了。

那我们继续修改代码:

for(vector<A*>::iterator iter=vec.begin(); iter!=vec.end(); iter++)
{
    if(5 == (*iter)->id)
   {
        delete *iter;
        iter = veci.erase(iter);
    }
}

erase返回值是这样的:An iterator that designates the first element remaining beyond any elements removed, or a pointer to the end of the vector if no such element exists

仔细阅读上面的代码实际上也有问题,首先代码不能删除连续两个为5的元素,因为删除第一个之后,iter指向第二个,自增后,就指向了第二个后面了;其次如果元素5位于vector最后,删除后iter自增也会出错。

那么正确的写法可以如下:

for(vector<A*>::iterator iter=vec.begin(); iter!=vec.end(); )
{
    if(5 == (*iter)->id)
   {
        delete *iter;
        iter = veci.erase(iter);
    }
    else
    {
        iter++;
    }
}

这样就可以解决连续两个相同元素的问题,同时将要删除的元素位于最后的话,也不会有问题,因为删除后,erase返回vector.end()。

vector删除操作 erase方法注意事项

标签:vector   stl   删除   erase   内存释放   

原文地址:http://jincheng.blog.51cto.com/4625177/1618279

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!