在任一方向上迭代容器并擦除元素

Question

我有两个 std::maps:

std::map<int,int> map1;
std::map<int,int> map2;

我需要一个向后迭代，另一个向前迭代（因为这是数据访问的模式）。在迭代时，我希望能够擦除元素继续迭代。

我想用同样的方法。

我已经看到使用模板展示如何进行双向迭代的示例，但没有演示擦除元素（这很重要，因为 erase() 仅适用于前向迭代器）：

Iterating over a container bidirectionally

我已经看到了 reverse_iterator 示例，它们可以擦除，但它们不是双向的：

How to call erase with a reverse iterator using a for loop

但我想双向迭代并擦除？

Answer 1

您可以为std::map、std::set 或std::list 编写函数：

template<typename Cont, typename Pred>
void bidir_remove_if( Cont &c, Pred p, bool forward )
{
    auto b = c.begin();
    auto e = c.end();
    while( b != e ) {
       auto it = forward ? b++ : --e;
       if( p(*it) ) {
           auto end = b == e;
           ( forward ? it : e ) = c.erase( it );
           if( end ) break;
       }
    }
}

live example

注意 - 由于迭代器无效，您不能将此函数用于 std::vector（并且您最好不要使用擦除删除习语）。