Question

我做了一个简单的函数，需要2 std::array并返回另一个std::array，其中包含std::arrays s中值的平均值。

template<typename T, std::size_t N>
std::array<T, N> average(const std::array<T, N>& a1, const std::array<T, N>& a2)
{
    std::array<T, N> array;
    for (std::size_t i = 0; i < N; ++i)
        array[i] = (a1[i] + a2[i]) / 2;

    return array;
}

完美无缺。现在我想计算 N 向量的平均值。所以我做了这个

template<typename T, std::size_t N, typename... Ts>
std::array<T, N> average(const Ts&... args)
{
    std::array<T, N> result;
    for (std::size_t i = 0; i < N; ++i)
    {
        T addedValues = 0;
        for (const auto& array : { args... })
            addedValues += array[i];

        result[i] = addedValues / sizeof...(args);
    }

    return result;
}

哪个也有效，但我必须指定生成的模板参数

std::array<int, 3> a{ 1, 2, 3 };
std::array<int, 3> b{ 3, 4, 5 };

auto c = average<int, 3>(a, b); //'<int, 3>' not good, possible without?

我想不出另一种方式，有人可以帮我吗？

Answer 1

前提是您还在模板功能的主体中进行了必要的更改。您可以按如下方式更改模板功能的签名：

template<typename T, std::size_t N, typename... Ts>
std::array<T, N> average(std::array<T, N> const &arr, const Ts&... args)

这样您就不必明确指定模板参数，模板参数推导会推导出T和N，因此您可以将average模板函数称为：

std::array<int, 3> a{ 1, 2, 3 };
std::array<int, 3> b{ 3, 4, 5 };

auto c = average(a, b);

使用模板计算向量的平均值

1 个答案: