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()。
时间: 2024-10-06 02:54:01