遍历 std::multimap 并删除某些条目

Question

我想遍历 std::multimap 中的所有项目（所有键的所有值），并删除所有满足某些条件的条目：

#include <map>

typedef int KEY_TYPE;
typedef int VAL_TYPE;

bool shouldRemove(const KEY_TYPE&, const VAL_TYPE&);

void removeFromMap(std::multimap<KEY_TYPE,VAL_TYPE>& map){
    for (auto it = map.begin(); it != map.end(); it++){
        if (shouldRemove(it->first,it->second))
            map.erase(it);
    }
}

除非第一个项目被删除，否则迭代将起作用，然后引发以下错误：

map/set 迭代器不可递增

如何重写removeFromMap 函数才能正常工作？该代码应适用于地图的各种键值类型。

我正在使用 C++ 11 和 Visual Studio 2013。

Answer 1

您需要在进行擦除之前增加您的迭代器。当您执行 map.erase(it); 时，迭代器 it 变得无效。但是，地图中的其他迭代器仍然有效。因此，您可以通过对迭代器进行后增量来解决此问题...

auto it = map.begin();
const auto end = map.end();

while (it != end)
{
    if (shouldRemove(it->first,it->second))
    {
        map.erase(it++);
                 // ^^ Note the increment here.
    }
    else
    {
       ++it;
    }
}

在map.erase() 参数内应用到it 的后增量将确保it 在项目被删除后保持有效，方法是在删除前递增迭代器以指向映射中的下一个项目。

map.erase(it++);

...在功能上等同于...

auto toEraseIterator = it;    // Remember the iterator to the item we want to erase.
++it;                         // Move to the next item in the map.
map.erase(toEraseIterator);   // Erase the item.

正如@imbtfab 在 cmets 中指出的那样，您也可以使用 it = map.erase(it) 在 C++11 中做同样的事情，而无需后递增。

另请注意，for 循环现在已更改为 while 循环，因为我们手动控制迭代器。

此外，如果您希望使 removeFromMap 函数尽可能通用，则应考虑使用模板参数并直接传递迭代器，而不是传递对多映射的引用。这将允许您使用任何地图样式的容器类型，而不是强制输入 multimap。

例如

template <typename Iterator>
void removeFromMap(Iterator it, const Iterator &end){
    ...
}

这也是标准 C++ <algorithm> 函数的执行方式（例如 std::sort(...)）。