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都适用，可依自己喜好选择。

From:http://www.cnblogs.com/ggjucheng/archive/2012/01/03/2310973.html

iterator erase( iterator pos );

对C++ STL中迭代器的删除需要慎重，如上方法。

1、vector

vector在遍历中进行 erase(pos) ，这个时候iter++的时候会出现bug，会跳过一个字段或者coredump。

解决方法：（要删除的时候直接赋值返回值，但是不能++）

vector<int>::iterator iter2;  
for(iter2 = vec.begin(); iter2 != vec.end(); )  
{  
    printf("%d " , *iter2 );     //注意这里  

    if( *iter2 == 3 ){  
        iter2 = vec.erase( iter2 );  
    }else{  
        iter2++;  
    }  
}  

2、list

list在遍历中进行 erase(pos) ，这个时候iter++的时候会出现bug，会直接coredump。

解决方案：

list<int>::iterator iter2;  
for(iter2 = vec.begin(); iter2 != vec.end(); iter2++)  
{  
    printf("%d " , *iter2 );  
    if( *iter2 == 3 ){  
        vec.remove( *iter2 );  //注意这里,调用remove( const TYPE &val );  
    }  
}  

3、map

list在遍历中进行 erase(pos) ，这个时候iter++的时候会出现bug，会出现未知问题。

解决方案：

map<int,int>::iterator iter;  
for(iter = testMap.begin(); iter != testMap.end(); iter++)  
{  
    printf("%d " , iter->second );  
    if( iter->second == 3 )  
    {  
        testMap.erase( iter->first ); //注意这里，是通过key删除  
    }  
}  

map<int,int>::iterator iter;
for(iter = testMap.begin(); iter != testMap.end(); iter++)
{
printf("%d " , iter->second );
if( iter->second == 3 )
{
testMap.erase( iter->first ); //注意这里，是通过key删除
}
}

From:http://blog.csdn.net/ghosc/article/details/7008521