1. 程式人生 > >STL Vector remove()和erase()的使用

STL Vector remove()和erase()的使用

STL中remove()只是將待刪除元素之後的元素移動到vector的前端,而不是刪除。若要真正移除,需要搭配使用erase()。
例子如下:
[cpp] view plain copy

#include <iostream>  
#include <vector>  
#include <algorithm>  
#include <functional>  

using namespace std;  

int main()  
{  
    vector<int> Numbers;   //vector containing numbers  

    Numbers.push_back(10);  
    Numbers.push_back(20);  
    Numbers.push_back(10);  
    Numbers.push_back(15);  
    Numbers.push_back(12);  
    Numbers.push_back(7);  
    Numbers.push_back(9);  

    vector<int>::iterator  it;  

    cout << "Before calling remove" << endl << "Numbers { " ;  

    for(it = Numbers.begin(); it != Numbers.end(); it++)  
    {  
        cout << *it << "\t" ;  
    }  
    cout << " }\n" << endl ;  

    // remove all elements from Numbers that match 10  
    vector<int>::iterator ret = remove(Numbers.begin(), Numbers.end(), 10) ;  

    cout << "After calling remove" << endl<< "Numbers { " ;  
    for(it = Numbers.begin(); it != Numbers.end(); it++)  
    {  
        cout << *it << "\t" ;  
    }  
    cout << " }\n" << endl ;  

    cout << "Total number of elements removed from Numbers = "  
        << Numbers.end() - ret << endl ;  

    cout << "Valid elements of Numbers { " ;  
    for(it = Numbers.begin(); it != ret; it++)  
    {  
        cout << *it << "\t" ;  
    }  
    cout << " }\n" << endl ;  

}  

這裡寫圖片描述
我們可以看出:
對於原vector { 10 20 10 15 12 7 9 },刪除10,會將10後面的元素移動到前面
(注意以下逐個元素對齊,模擬元素在記憶體中的位置,這樣就容易看出變化規律)
原vector
10 20 10 15 12 7 9
遇到第一個10,陣列變成
20 10 15 12 7 9
遇到第二個10移動到7 9之後又遇到第三個10於是剩下記憶體中未移動的7 9
20 15 12 7 9 7 9
遇到第三個10
20 15 12 7 9 7 9
因此,remove()需要和erase()配合使用
svec.erase(remove(svec.begin(),svec.end(),”be”), svec.end());
這句的意思是,取得”be”的位置(位於結尾),然後刪除”be”到原vector結尾的所有元素

[cpp] view plain copy

iterator的使用陷阱:  
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);  
}  

這樣使用是錯誤的,因為earase結束後,iter變成了野指標,iter++就產生了錯誤。
erase()返回值是一個迭代器,指向刪除元素下一個元素;如果是刪除某範圍內的元素時:返回值也表示一個迭代器,指向最後一個刪除元素的下一個元素;

[cpp] view plain copy

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

這樣無法刪除連續兩個3.
需要寫成

[cpp] view plain copy

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