[1 容器] 9. 慎重选择删除元素的方法

1 删除容器中的指定元素

假定STL容器c，需要删除c中所有值为1963的元素。

Container<int> c;

注意，完成这一任务的方法随容器类型而异。

(1) 连续内存容器vector，deque或string

使用erase-remove习惯用法：

// remove移动指定区间中的元素知道所有"不删除"的元素在区间的开头
c.erase(remove(c.begin(), c.end(), 1963), c.end());

(2) list

c.remove(1963);

(3) 标准关联容器set，multiset，map或multimap

c.erase(1963);

时间复杂度对数时间。序列容器基于remove的方法，时间复杂度线性时间。

2 删除指定判别式返回true的每一个对象

判别式：

bool badValue(int x);

(1) 连续内存容器vector，deque或string

// remove移动指定区间中的元素知道所有"不删除"的元素在区间的开头
c.erase(remove_if(c.begin(), c.end(), badValue), c.end());

 (2) list

c.remove_if(badValue);

(3) 标准关联容器set，multiset，map或multimap

[1] 简单但效率较低的方法

利用remove_copy_if把我们需要的值复制到一个新容器中，然后把新容器和原来容器中的内容交换：

AssocContainer<int> c;
...
// 保存不被删除的值的临时容器
AssocContainer<int> goodValues;
// 把不被删除的值从c复制到goodValues中
remove_copy_if(c.begin(), c.end(), inserter(goodValues, goodValues.end()), badValue);
// 交换c和goodValues
c.swap(goodValues);

这种方法的缺点是需要复制所有不被删除的元素。

[2] 直接从原始容器中删除元素，提高效率

因为关联容器没有提供类似remove_if的成员函数，所以我们必须写一个循环来遍历c中的元素，并在遍历过程中删除元素。

很多程序员首先想到的是：

AssocContainer<int> c;
...
for (AssocContainer<int>::iterator i = c.begin(); i!=c.end(); ++i) {
    if (badValue(*i)) {
        c.erase(i);
    }
}

上述代码有bug，会导致不确定的行为。当容器中的一个元素被删除时，指向该元素的所有迭代器都将变得无效。一旦c.erase(i)返回，i就成为无效值。

为了避免这个问题，我们要确保在调用erase之前，有一个迭代器指向c中的下一个元素。最简单的办法是，调用时对i使用后缀递增：

AssocContainer<int> c;
...
for (AssocContainer<int>::iterator i = c.begin(); i!= c.end(); ) {
    if (badValue(*i)) {
        c.erase(i++);
    } else {
        ++i;
    }
}

表达式i++的值是i的旧值，作为副作用，i被递增。我们把旧的i传给erase，在erase开始执行前递增了i，i是有效的。

3 不仅要删除使badValue返回true的元素，还要在每次删除元素时向日志写一条信息

(1) 标准关联容器set，multiset，map或multimap

比较简单，仅需要对刚才的循环做简单的修改：

ofstream logFile;
AssocContainer<int> c;
...
for (AssocContainer<int>::iterator i = c.begin(); i!=c.end(); ) {
    if (badValue(*i)) {
        logFile << "Erasing " << *i << "\n";
        c.erase(i++);
    } else {
        ++i;
    }
}

(2) 连续内存容器vector，deque或string

不能再使用erase-remove用法，因为没办法用erase或remove向日志文件写信息。

不能使用类似关联容器的循环，因为对于连续内存容器，调用erase不仅会使指向被删除元素的迭代器无效，也会使被删除元素之后的所有迭代器失效。

对于连续内存容器，要利用erase的返回值，erase的返回值指向紧随被删除元素的下一个元素的有效迭代器。

for (SeqContainer<int>::iterator i = c.begin(); i != c.end();) {
    if (badValue(*i)) {
        logFile << "Erasing " << *i << "\n";
        // erase的返回值赋值给i,使i有效
        i = c.erase(i);
    } else {
        ++i;
    }
}

(3) list

惯例是对list采取和连续内存容器相同的方式