安全删除STL容器元素

STL容器迭代过程中删除元素技巧(转)

1.连续内存序列容器(vector,string,deque)

序列容器的erase方法返回值是指向紧接在被删除元素之后的元素的有效迭代器，可以根据这个返回值来安全删除元素。

vector<int> c;
for(vector<int>::iterator it = c.begin(); it != c.end();)
{
if(need_delete())
  it = c.erase(it);
 else
  ++it; 
}

 

2.关联容器(set,multiset,map,multimap)

关联容器的erase方法没有返回值，被删除的迭代器失效，所以删除前必须确保能得到下一个迭代器，可以用“后置递增迭代器”技术。

map<int,int> m;
for(map<int,int>::iterator it = m.begin(); it != m.end();)
{
 if(need_delete()) 
  m.erase(it++);
 else
  ++it;
}

m.erase得到it的一个副本，在erase真正开始之前it已经递增了。

所以erase得到了当前的迭代器，在erase内部工作开始之前it已经++了，正好满足我们的需要。

 

3.非连续内存序列容器(list)

只所以单独列出来是因为以上两种方法对list都适用，可依自己喜好选择。