在C ++ 14中对此C ++ 17折叠表达式有什么好的替代方法?

时间:2018-06-23 23:45:30

标签: c++ templates c++14 variadic-templates fold-expression

这是C ++ 17中基于lambda的一个很好的,简洁的折叠表达式:

#include <cstdint>

using ::std::uint64_t;

constexpr auto sumsquares = [](auto... n) { return ((n * n) + ...); };

// I want this to work.
uint64_t foo(uint64_t x, uint64_t y, uint64_t z)
{
    return sumsquares(x, y, z);
}
// And this too
double bar(uint64_t x, double y)
{
    return sumsquares(x, y);
}

我已经编写了这段代码,以在C ++ 14中执行类似的操作,但是似乎比它应该的更加冗长和混乱。我正在寻找一种以相对清晰简洁的方式在C ++ 14中表达上述C ++ 17代码的方法。确切地说,我希望能够编写类似函数调用的语法,以计算一些已知维数的矢量的矢量幅度平方。尺寸的数量可以任意变化。坐标系统各个组成部分的精确数字类型也可以是任意的,也可能是异类的。但是在C ++ 14中处理C ++ 17折叠表达式的一般方法是理想的。

#include <cstdint>
#include <utility> 

using ::std::uint64_t;

namespace {
    static constexpr struct {
        template <typename T>
        auto operator()(T && n) const
        {
           return n*n;
        }
        template <typename T, typename... S>
        auto operator()(T && n, S && ... s) const
        {
            return (n * n) + (*this)(::std::forward<S>(s)...);
        }
    } sumsquares;
}

// I want this to work.
uint64_t foo(uint64_t x, uint64_t y, uint64_t z)
{
    return sumsquares(x, y, z);
}
// And this too
double bar(uint64_t x, double y)
{
    return sumsquares(x, y);
}

3 个答案:

答案 0 :(得分:5)

#include <utility>
#include <functional>

template<class F, class A0>
auto fold(F&&, A0&& a0) {
    return std::forward<A0>(a0);
}

template<class F, class A0, class...As>
auto fold(F&& f, A0&&a0, As&&...as) {
    return f(std::forward<A0>(a0), fold(f, std::forward<As>(as)...));
}

auto sum_squares=[](auto&&...args) {
    return fold(std::plus<>{}, (args * args)... );
};

答案 1 :(得分:4)

以下一种可能的写sum()的非递归方式

template <typename ... Ts>
auto sum (Ts ... ts)
 {
   using unused = int[];

   std::common_type_t<Ts...>  ret {};

   (void)unused{ 0, (ret += ts, 0)... };

   return ret;
 }

使用sum()函数,sum_of_squares()可以简单地编写如下

template <typename ... Ts>
auto sum_of_squares (Ts ... ts)
 { return sum( (ts*ts)... ); }

答案 2 :(得分:3)

是的,实际上,您可以通过在多个对象上定义一个通用的和,来做得更简单,更通用:

template<typename T>
T sum(T x)
{
    return x;
}
template<typename U, typename... T>
auto sum(U x, T... nums)
{
    return x + sum(nums...);
}

template<typename... T>
auto sum_of_squares(T... nums)
{
    auto square = [](auto x) { return x * x; };
    return sum(square(nums)...);
}

Live demo herehere,您可以看到生成的程序集(似乎是最佳的)。尽管我个人确实不喜欢将auto作为返回类型,但是在这里它可以避免使用非常复杂的表达式来获取该类型。