Sweet Snippet系列 之 元素删除

1. 引子:

  平时代码总会遇到一些关于集合的操作，例如添加，排序等等，都可算作稀松平常，但是集合涉及的删除操作却一直有个大坑，我自己便跳进去过好几回，在此简单一记，以自警警人～

 

  2. 问题:

  删除整数容器中的偶数

 

  问题很简单，似乎一个简单的for循环就搞定了，就像这样：

#include <vector>
#include <iostream>

using namespace std;

int main() {
    vector<int> vec = { 1, 2, 2, 3 };

　　auto isEven = [](int value) -> bool { 
　　    return (value & 1) == 0; 
　　};
　　
    auto output = [&vec]() {
        for (auto v : vec) {
            cout << v << " ";
        }
        cout << "\n";
    };

    for (auto iter = vec.begin(); iter != vec.end(); ++iter) {
        if (isEven(*iter)) {
            vec.erase(iter);
        }
    }

    output();

    return 0;
}

代码虽易，可惜明眼人一看就知道问题所在了：erase操作会影响iterator的正常迭代！所以上面看似简洁的代码并不能产生正确的结果（就上面集合来说，我们会遗漏处理第三个元素，即2），我们必须采用另外的方法：

一种就是使用buffer（稍有些冗余）：

// omit codes ...

    {
        vector<int> buffer;
        for (auto iter = vec.begin(); iter != vec.end(); ++iter) {
            if (!isEven(*iter)) {
                buffer.push_back(*iter);
            }
        }
        vec.swap(buffer);
    }

    // omit codes ...

近来还看到一种使用goto的tricky 方案（当然，个人不是很喜欢）：

 // omit codes ...

 restart:
    for (auto iter = vec.begin(); iter != vec.end(); ++iter) {
        if (isEven(*iter)) {
            vec.erase(iter);
            goto restart;
        }
    }
   
    // omit codes ...

总的来说，我最认可的还是下面这种写法（平心而论，确实不及for形式简洁）：

 // omit codes ...
 
  auto iter = vec.begin();
    while (iter != vec.end()) {
        if (isEven(*iter)) {
            iter = vec.erase(iter);
        }
        else {
            ++iter;
        }
    }  
    
   // omit codes ...

不知还有什么另外的方法，有兴趣的朋友可以告知一下 ：）

 

OK，That’s all ~