vector中erase用法注意事項

以前就發現了vector中的erase方法有些詭異(^_^),稍不注意,就會出錯。今天又一次遇到了,就索性總結一下,尤其是在循環體中用erase時,由於vector.begin() 和vector.end()是變化的,因此就引入了錯誤的可能性。

vector<int> veci;
veci.push_back(1);
veci.push_back(2);
veci.push_back(3);
veci.push_back(4);
veci.push_back(5);
veci.push_back(3);
veci.push_back(2);
veci.push_back(3);

for(vector<int>::iterator iter=veci.begin(); iter!=veci.end(); iter++)
{
      if( *iter == 3)
             veci.erase(iter);
}

乍一看這段代碼,很正常。其實這裏面隱藏着一個很嚴重的錯誤:當veci.erase(iter)之後,iter就變成了一個野指針,對一個野指針進行 iter++ 是肯定會出錯的。

查看MSDN,對於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,於是改代碼:

for(vector<int>::iterator iter=veci.begin(); iter!=veci.end(); iter++)
{
      if( *iter == 3)
             iter = veci.erase(iter);
}

這段代碼也是錯誤的:1)無法刪除兩個連續的"3"; 2)當3位於vector最後位置的時候,也會出錯(在veci.end()上執行 ++ 操作)

正確的代碼應該爲:

for(vector<int>::iterator iter=veci.begin(); iter!=veci.end(); )
{
     if( *iter == 3)
          iter = veci.erase(iter);
      else
            iter ++ ;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章