Sweet Snippet系列之元素删除

用户2615200

发布于 2018-08-02 17:15:00

2170

发布于 2018-08-02 17:15:00

文章被收录于专栏：tkokof 的技术,小趣及杂念tkokof 的技术,小趣及杂念

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 ~

本文参与腾讯云自媒体分享计划，分享自作者个人站点/博客。

原始发表：2014年04月08日，如有侵权请联系 cloudcommunity@tencent.com 删除

其他

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体分享计划，欢迎热爱写作的你一起参与！

其他

登录后参与评论

0 条评论

热度