STL Vector remove()和erase()的使用
阿新 • • 發佈:2018-12-31
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 ++ ;
}