在容器中使用erase函數，迭代器的處理

在c++編程中，用到迭代器的時候，往往不知道如何刪除當前迭代器指向的元素。

erase函數：

  返回下一個迭代器。

 

只使用vector的erase函數，記住，該函數是迭代器失效，返回下一個迭代器。

#include <iostream>
#include <vector>
using namespace std;
 
int main()
{
    vector<int> arr;
    arr.push_back(6);
    arr.push_back(7);
    arr.push_back(8);
    arr.push_back(9);

    for(vector<int>::iterator it=arr.begin(); it!=arr.end(); )
    {
        if(* it == 8)
        {
            it = arr.erase(it); //不能寫成arr.erase(it);
        }
        else
        {
            ++it;
        }

 

看下面的一個程序，刪除值為4和7的元素，為什么只刪除了4？

#include<iostream>
#include<string>
#include<vector>
using namespace std;
int main()
{

    vector<int> iVec;

    vector<int>::iterator it;
    int i;

    for( i=0;i<10;i++)

        iVec.push_back(i);

    for( i=0;i<iVec.size();i++)
        cout<<iVec[i]<<ends;
    cout<<endl;

    for(it=iVec.begin();it!=iVec.end();++it)

    {

        if(*it ==4 || *it == 7)

        {

            it=iVec.erase(it);

        }

        else

            ++it;

    }

   for( i=0;i<iVec.size();i++)
        cout<<iVec[i]<<ends;
    cout<<endl;

}

沒有刪除7，為什么？

l例如上面這個 1 2 3 4 5 6 7 8 9

如果要刪除4 和7 兩個元素

則刪除4后iterator返回5，不滿足if()判斷，這時++it執行兩次，一次是else里面，一次是for()循環里面，導致7被錯過

改為：

 it=iVec.erase(it);

      --it;//這里回退一個

上面的程序基本上屬於弱智型程序，沒必要it++，使用下面的：

    for(it=iVec.begin();it!=iVec.end();++it)

    {

        if(*it ==4 || *it == 7)

        {

            it=iVec.erase(it);

        }

     

    }

 （2014-12-1修改：

上面程序經過網友指出if(*it ==4 || *it == 5)有問題，確實如此，2個相鄰的元素，在for++中被跳過了。為了處理這種情況，把for循環中的it++移動到for中。

for(it=iVec.begin();it!=iVec.end();)

{

　　if(*it==4 || *it==5)

　　　　it=iVec.erase(it);

　　else

　　　　it++;

}

這樣程序就可以處理4,5,4,7之類的了。）

 

 

不用循環結合remove使用：

1、remove並不是刪除，僅僅是移除，要加上erase才能完成刪除。 
2、remove並不是刪除指定位置的元素，而移除所有指定的元素。 
3、用algorithm代替成員函數不是一個好的選擇。

remove只是簡單地用"未刪除"對象來填補被刪除對象留下的缺口,每一個刪除對象在尾部還是會留下一個相應的"死亡"對象. 
可以這樣來使用remove... 
vector<int>vec; 
vec.erase(remove(vec.begin(),vec.end(),value),vec.end()); //value是要刪除的值

 

    
vector<int> v;
     
    int i;
    for( i=0;i<10;i++)

        v.push_back(i);

remove(v.begin(),v.end(),5);
    remove(v.begin(),v.end(),6);
    for(vector<int>::iterator it=v.begin();it!=v.end();it++)
    {
         
        cout<<*it<<ends;
    }

輸出：

0123478999

 

 

給一個應用列子：

有 $n$ 個小朋友做游戲，他們的編號分別是 $1,2,3...n$。他們按照編號從小到大依次順時針圍成一個圓圈，從第一個小朋友開始從 $1$報數，依次按照順時針方向報數(加一)，報 $m$ 的人會離開隊伍，然后下一個小朋友會繼續從 $1$ 開始報數，直到只剩一個小朋友為止。

輸入格式

第一行輸入倆個整數，$n$，$m$。（$1\le n,m\le 1000$）

輸出格式

輸出最后一個小朋友的編號，占一行。

樣例輸入

10 5

樣例輸出

3

#include<bits/stdc++.h>
using namespace std;
vector<int>a;
int main()
{
  int n,m;
  cin>>n>>m;
  for(int i=1;i<=n;i++)
    a.push_back(i);
  int now=1;
  vector<int>::iterator it;
  while(a.size()>1)
  {
      for(it=a.begin();it!=a.end();)
      {
          if(now==m)
          {
              it=a.erase(it);
              now=1;
          }
          else
          {
             it++;
             now++;
          }

      }
  }
  cout<<a[0]<<endl;
  return 0;
}