STL中map/vector的刪除元素操作
阿新 • • 發佈:2019-01-02
在我們使用C++中的STL的時候,可以使用迭代器iterator進行遍歷,但是當我們通過iterator對vector和map刪除元素的時候,要格外的小心,往往操作不當,導致iterator失效,後果就是程式奔潰。
1. 對於vector,erase會返回下一個iterator。所以一般採用的方法是:
因為在使用erase的時候,刪除元素前面的iterator有效,但是後面的iterator就不可預知了。
執行結果:#include <iostream> #include <vector> #include <iterator> using namespace std; void displayAllDate(vector<int>& inputDate) { vector<int>::iterator itor = inputDate.begin(); cout << endl; while (itor != inputDate.end()) { cout << *itor << " "; ++itor; } } int main() { vector<int> inputDate; for (int i = 0; i < 10; i++) { inputDate.push_back(i); } vector<int>::iterator iter = inputDate.begin(); while (iter != inputDate.end()) { if (0 == (*iter)%2) { iter = inputDate.erase(iter); } else { ++iter; } } displayAllDate(inputDate); return 0; }
2. 對於map,刪除erase之後,只會影響到當前的iterator,因此我們一般推薦以下方法:
執行結果:#include <iostream> #include <map> #include <iterator> using namespace std; void displayAllDate(map<int, int>& inputDate) { map<int, int>::iterator itor = inputDate.begin(); cout << endl; while (itor != inputDate.end()) { cout << itor->second << " "; ++itor; } } int main() { map<int, int> inputDate; for (int i = 0; i < 10; i++) { inputDate[i] = i + 10; } map<int, int>::iterator iter = inputDate.begin(); while (iter != inputDate.end()) { if (0 == (iter->second) % 2) { inputDate.erase(iter++); } else { ++iter; } } displayAllDate(inputDate); return 0; }
但是要注意的一點,map的erase操作在window下面也可以用vector的方法實現,但是在linux下是編譯不過的。所以推薦使用上面的方法。
執行結果是:#include <iostream> #include <map> #include <iterator> using namespace std; void displayAllDate(map<int, int>& inputDate) { map<int, int>::iterator itor = inputDate.begin(); cout << endl; while (itor != inputDate.end()) { cout << itor->second << " "; ++itor; } } int main() { map<int, int> inputDate; for (int i = 0; i < 10; i++) { inputDate[i] = i + 10; } map<int, int>::iterator iter = inputDate.begin(); while (iter != inputDate.end()) { if (0 == (iter->second) % 2) { iter = inputDate.erase(iter); } else { ++iter; } } displayAllDate(inputDate); return 0; }
參考: