vector map删除元素

一般，我们都是通过迭代器iterator对容器元素进行遍历。

如果需要删除其中的某些元素，需要怎样操作呢？ 显然直接erase会导致iterator出错。

对于vector，可以借助erase函数的返回值为“the new location of the element that followed the last element erased by the function call”。

下面是一个实例

void vector_delete_test()
{
std::vector<int> v;
v.push_back(1);
v.push_back(3);
v.push_back(5);
v.push_back(8);
v.push_back(2);


std::vector<int>::iterator it;
for(it = v.begin(); it != v.end();)
{
           if(*it > 2)
          {
               it = v.erase(it); 

         //refer to http://www.cplusplus.com/reference/vector/vector/erase/
           }
       else
       {
        it++;
       }
}


for(it = v.begin(); it != v.end(); it++)
{
printf("%d \n", *it);
}
}

对于map,  可以考虑如下方法
void map_delete_test()
{
std::map<int, int> map;
std::map<int, int>::iterator it;


map[1] = 10;
map[2] = 20;
map[3] = 30;
map[4] = 40;


for(it = map.begin(); it != map.end();)
{
  if(it->second == 30)
  {
   // std::map<int,int>::iterator tmp = it;
   // it++;
   // map.erase(tmp);

  map.erase(it++); 

                 //利用后++的特性

                 //这里首先把it的值传给函数(传参)，然后执行++操作，最好执行erase函数
  }
  else
  {
   it++;
  }
}

for(it = map.begin(); it != map.end();)
{
 printf("%d \n", it->second);
 it++;
}
}

int main()
{
   /vector_delete_test();

    map_delete_test();
   

   return 0;
}