将递归转换为循环

时间:2016-04-09 08:32:32

标签: c++ recursion optimization

我有一个未知类型T可能是非副本或可移动分配的函数T op(const T &foo, Other bar),它根据现有函数计算并返回一个新的T,并且递归函数:

template<typename Iter, typename T>
T foldLeft(Iter first, Iter last, const T &z, std::function<T(T, Other)> op)
{
    if (first == last) {
        return z;
    }
    return foldLeft(std::next(first), last, op(z, *first), op);
}

编译器不能总是优化尾调用,因为T可能有一个非平凡的析构函数。我正在尝试用循环手动重写它,但无法弄清楚如何重新分配到z

1 个答案:

答案 0 :(得分:5)

您可以执行以下操作,但限制很奇怪,使用std::accumulate似乎更简单

template<typename Iter, typename T, typename Fn>
T foldLeft(Iter first, Iter last, const T &z, Fn op)
{
    std::aligned_storage_t<sizeof (T), alignof (T)> buf;
    T* res = new (&buf) T(z);
    for (auto it = first; it != last; ++it) {
        auto&& res2 = op(res, it);
        res->~T();
        res = new (&buf) T(std::move(res2));
    }
    T final_res = *res;
    res->~T();
    return final_res;
}

请注意,它不是例外安全。