使用back_inserter()或inserter()提高std :: copy()的效率

时间:2012-09-08 00:56:43

标签: c++ inserter

back_inserterinsert_iterator非常方便,但效率也很低!

例如,当您追加char时,当copy时,每个元素都会产生大量开销,而在许多情况下,确实不需要是。

有没有办法让它们更有效率?

1 个答案:

答案 0 :(得分:1)

是的,您可以定义可以劫持可优化呼叫的新版std::copy。 :)

以下是Visual C ++和GCC的示例(或“hack”,如果您希望看到半空的玻璃)。

在我的个人计算机上(我使用VC ++ 2010),下面的代码可以更快地调用 十次 ! GCC的基准也在这里,差异为5倍:old versionnew version

之前你使用它:

请注意,此代码假设容器提供了类似vector的界面

正如目前所写,这仅适用于C ++ 11,因为它使用type_traits标头的元编程功能优化复制操作将保持异常安全的情况

如果您不需要异常安全(尽管您在实际执行此操作之前应该三思而行),或者如果您有其他方法来检查此类数据类型,那么您可以更改

typename enable_if<..., typename insert_iterator<C> >::type

为:

insert_iterator<C>

并且其余代码也适用于C ++ 03。

namespace std
{
    template<class FwdIt, class C>
    back_insert_iterator<C> copy(
        FwdIt begin, FwdIt end, back_insert_iterator<C> it,
        forward_iterator_tag * =
          static_cast<typename iterator_traits<FwdIt>::iterator_category *>(0))
    {
        struct It : public back_insert_iterator<C>
        {
            using back_insert_iterator<C>::container;
            static C &deref(C &c) { return  c; }
            static C &deref(C *c) { return *c; }
        };
        copy(begin, end, inserter(It::deref(static_cast<It &>(it).container),
                      It::deref(static_cast<It &>(it).container).end()));
        return it;
    }

    template<class FwdIt, class C>
    typename enable_if<  // Only do this if it would be exception-safe!
        is_nothrow_copy_constructible<typename C::value_type>::value &&
        is_nothrow_copy_assignable<typename C::value_type>::value,
        insert_iterator<C>
    >::type copy(
        FwdIt const &begin, FwdIt const &end,
        insert_iterator<C> output,
        forward_iterator_tag * =                  // only forward iterators
          static_cast<typename iterator_traits<FwdIt>::iterator_category *>(0))
    {
        struct It : public insert_iterator<C>
        {
            using insert_iterator<C>::container;  // protected -> public
            using insert_iterator<C>::iter;       // protected -> public
            static C &deref(C &c) { return  c; }
            static C &deref(C *c) { return *c; }
        };
        It &it(static_cast<It &>(output));
        typename C::iterator it_iter_end(It::deref(it.container).end());
        {
            // Convert iterators to offsets
            typename C::size_type const iter_end_off =
                std::distance(It::deref(it.container).begin(), it_iter_end);
            typename iterator_traits<typename C::iterator>::difference_type off
                = std::distance(It::deref(it.container).begin(), it.iter);

            // Resize container
            It::deref(it.container).resize(
                It::deref(it.container).size() +
                static_cast<typename C::size_type>(std::distance(begin, end)));

            // Renormalize, in case invalidated
            it.iter = It::deref(it.container).begin();
            std::advance(it.iter, off);
            it_iter_end = It::deref(it.container).begin();
            std::advance(it_iter_end, iter_end_off);
        }
        typename C::iterator result
          = copy_backward(it.iter, it_iter_end, It::deref(it.container).end());
        copy_backward(begin, end, result);
        return inserter(It::deref(it.container), result);
    }
}