vector删除指定元素

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>

using namespace std;

int main()
{
    vector<int> vs;
    for(int i = 0; i < 5; i++) vs.push_back(i);

    // 方法1:使用earse函数
    cout << "删除指定元素:3" << endl;
    for(vector<int>::iterator it = vs.begin(); it != vs.end(); it++)
    {
        // if(*it == 3) vs.erase(it);
        if(*it == 3) it = vs.erase(it);
    }

    for(int i = 0; i < vs.size(); i++) cout << vs[i] << " ";
    puts("");

    // 方法2:使用swap到最后再pop_back
    cout << "删除指定元素:1" << endl;
    for(int i = 0; i < vs.size(); i++)
    {
        if(vs[i] == 1)
        {
            swap(vs[i], vs[vs.size() - 1]);
            vs.pop_back();
        }
    }

    for(int i = 0; i < vs.size(); i++) cout << vs[i] << " ";
    puts("");

    return 0;
}