STL中容器元素的删除

STL

STL中的容器可以分为两类：一类是按照数组方式连续存储元素的容器，如vector、deque，另一类是按照不连续的节点存储元素的容器，如set、map、list等。
那么这两类容器删除元素的时候需要注意一个问题，使用iterator指定删除某个元素的时候，要注意返回的iterator的值。下面分别举set、vector的例子来说明。

set删除元素

假如要删除set中满足某个条件的元素，这时候需要遍历，看代码：

set<int> s;
s.insert(1);
s.insert(4);
s.insert(2);
s.insert(3);
for (set<int>::iterator it = s.begin(); it != s.end(); it ++)
    cout << *it << endl;
for (set<int>::iterator it = s.begin(); it != s.end(); )
{
    if (*it % 2 != 0) // 删除奇数
        s.erase(it++);
    else
        it ++;
}

大家注意到s.erase(it++)这一句，it++返回是+1之前的it的值，那么此时删除的正是满足条件的元素，而it本身又进行了+1，指向了下一个元素，那你可能会问为什么不直接删除之后再+1呢，如下代码：

s.erase(it);
it++;

这样为什么不行呢？因为在erase之前，it其实指向的还是原来的元素，只不过它已经不存在set中，那么此时it++，得到的值不是set中的那一个元素，而是不确定的，对这个指针进行访问，会得到意想不到的结果，甚至导致程序崩溃。

vector删除元素

vector内部存储元素是采用数组方式存储的，即是连接的，删除元素可以按照以下方法：

vector<int> ve;
ve.push_back(1);
ve.push_back(3);
ve.push_back(2);
ve.push_back(4);
for (vector<int>::iterator it = ve.begin(); it != ve.end();)
{
    if (*it % 2 != 0)
        ve.erase(it);
    else
        it ++;
}

vector跟set不一样的地方在于，调用erase之后，it自动指向下一个元素，所以在遍历删除的时候需要注意，erase之后不需要再进行it++，否则会跳过了一个元素。