Sweet Snippet系列之元素删除

矫情吗；* 2022-08-26 12:22 162阅读 0赞

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 ~