Sweet Snippet系列 之 元素删除

时间:2022-06-05
本文章向大家介绍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 ~