如何在迭代std :: set时从中删除元素

Question

我该如何在遍历 std::set 时删除其中的元素？

我的第一次尝试如下：

set<T> s;

for(set<T>::iterator iter = s.begin(); iter != s.end(); ++iter) {
    //Do some stuff
    if(/*some condition*/)
        s.erase(iter--);
}

但是，如果我们想从集合中删除第一个元素，iter-- 会使迭代器失效，这将带来问题。

有什么标准方法可以解决这个问题吗？

- dspyz

Answer 1

标准的做法是像这样：

for(set<T>::iterator iter = s.begin(); iter != s.end();)
{
   if(/*some condition*/)
   {
      s.erase(iter++);
   }
   else
   {
      ++iter;
   }
}

根据第一个条件，我们可以确定即使将iter的副本传递给erase函数，但由于我们的iter已在调用erase之前被递增过，因此它不会被非法化。

C++11中，代码将类似于：

for(set<T>::iterator iter = s.begin(); iter != s.end();)
{
   if(/*some condition*/)
   {
      iter = s.erase(iter);
   }
   else
   {
      ++iter;
   }
}

- ForEveR

---

第一段代码在Visual Studio中失败了，提示迭代器不兼容。无论如何，正如您在代码中指出的那样，std::erase返回新的迭代器。 - sajas

第二段代码对我有效。 - Larynx