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

fdsafwagdagadg6576發表於2016-06-21
REM

http://blog.csdn.net/ozwarld/article/details/7761519

STL中remove（）只是將待刪除元素之後的元素移動到vector的前端，而不是刪除。若要真正移除，需要搭配使用erase()。

例子如下：

[cpp] view plain copy
  1. // remove.cpp  
  2. // compile with: /EHsc  
  3. // Illustrates how to use the remove function.  
  4. //  
  5. // Functions:  
  6. //   remove - remove all elements from the sequence that match value.  
  7. //   begin - Returns an iterator that points to the first element in a  
  8. //           sequence.  
  9. //   end - Returns an iterator that points one past the end of a sequence.  
  10. //////////////////////////////////////////////////////////////////////  
  11.   
  12. // disable warning C4786: symbol greater than 255 character,  
  13. // okay to ignore  
  14. #pragma warning(disable: 4786)  
  15.   
  16. #include <iostream>  
  17. #include <vector>  
  18. #include <algorithm>  
  19. #include <functional>  
  20.   
  21. using namespace std;  
  22.   
  23.   
  24. int main()  
  25. {  
  26.     const int VECTOR_SIZE = 8 ;  
  27.   
  28.     // Define a template class vector of integers  
  29.     typedef vector<int > IntVector ;  
  30.   
  31.     //Define an iterator for template class vector of integer  
  32.     typedef IntVector::iterator IntVectorIt ;  
  33.   
  34.     IntVector Numbers(VECTOR_SIZE) ;   //vector containing numbers  
  35.   
  36.     IntVectorIt start, end, it, last;  
  37.   
  38.     start = Numbers.begin() ;   // location of first  
  39.                                 // element of Numbers  
  40.   
  41.     end = Numbers.end() ;       // one past the location  
  42.                                 // last element of Numbers  
  43.   
  44.     //Initialize vector Numbers  
  45.     Numbers[0] = 10 ;  
  46.     Numbers[1] = 20 ;  
  47.     Numbers[2] = 10 ;  
  48.     Numbers[3] = 15 ;  
  49.     Numbers[4] = 12 ;  
  50.     Numbers[5] = 7 ;  
  51.     Numbers[6] = 9 ;  
  52.     Numbers[7] = 10 ;  
  53.   
  54.   
  55.     cout << "Before calling remove" << endl ;  
  56.   
  57.     // print content of Numbers  
  58.     cout << "Numbers { " ;  
  59.     for(it = start; it != end; it++)  
  60.         cout << *it << " " ;  
  61.     cout << " }\n" << endl ;  
  62.   
  63.     // remove all elements from Numbers that match 10  
  64.      last = remove(start, end, 10) ;  
  65.   
  66.     cout << "After calling remove" << endl ;  
  67.   
  68.     // print content of Numbers  
  69.     cout << "Numbers { " ;  
  70.     for(it = start; it != end; it++)  
  71.         cout << *it << " " ;  
  72.     cout << " }\n" << endl ;  
  73.   
  74.     //print number of elements removed from Numbers  
  75.     cout << "Total number of elements removed from Numbers = "  
  76.         << end - last << endl ;  
  77.   
  78.     //print only the valid elements of Number  
  79.     cout << "Valid elements of Numbers { " ;  
  80.     for(it = start; it != last; it++)  
  81.         cout << *it << " " ;  
  82.     cout << " }\n" << endl ;  
  83.   
  84. }  


Output



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


[cpp] view plain copy
  1. 附註：  
  2. iterator的使用陷阱：  
  3. vector<int> veci;  
  4. veci.push_back(1);  
  5. veci.push_back(2);  
  6. veci.push_back(3);  
  7. veci.push_back(4);  
  8. veci.push_back(5);  
  9. veci.push_back(3);  
  10. veci.push_back(2);  
  11. veci.push_back(3);  
  12.   
  13.   
  14. for(vector<int>::iterator iter=veci.begin(); iter!=veci.end(); iter++)  
  15. {  
  16.       if( *iter == 3)  
  17.              veci.erase(iter);  
  18. }  




這樣使用是錯誤的，因為earase結束後，iter變成了野指標，iter++就產生了錯誤。
erase()返回值是一個迭代器，指向刪除元素下一個元素;如果是刪除某範圍內的元素時：返回值也表示一個迭代器，指向最後一個刪除元素的下一個元素;
[cpp] view plain copy
  1. for(vector<int>::iterator iter=veci.begin(); iter!=veci.end(); iter++)  
  2. {  
  3.       if( *iter == 3)  
  4.              iter = veci.erase(iter);  
  5. }  

這樣無法刪除連續兩個3.
需要寫成：
[cpp] view plain copy
  1. for(vector<int>::iterator iter=veci.begin(); iter!=veci.end(); )  
  2. {  
  3.      if( *iter == 3)  
  4.           iter = veci.erase(iter);  
  5.       else  
  6.             iter ++ ;  
  7. }  

相關文章