如何巧妙地将一些元素从std :: set移动到另一个容器?

时间:2012-01-05 21:40:58

标签: c++ algorithm stl

由于迭代器在删除操作时无效,如何在std::set的所有元素上使用最少量的代码进行迭代,删除其中的一些元素?

1 个答案:

答案 0 :(得分:6)

// Filters the elements of a set:  elements that satisfy the predicate 'pred'
// are removed from the source set and inserted into the output.
template <typename TSet, typename TOutputIterator, typename TPredicate>
void extract_if(TSet& s, TOutputIterator out, TPredicate pred)
{
    for (typename TSet::iterator it(s.begin()); it != s.end();)
    {
        if (pred(*it))
        {
            *out++ = *it;
            it = s.erase(it);
        }
        else
        {
            ++it;
        }
    }
}