c++ - 在编译时将std::array的每个元素相乘

我想将一个std::array转换为另一个std::array，将其每个元素乘以一个特定的数字。

我现在所拥有的显然不起作用:

#include <array>
#include <iostream>
#include <utility>

template <class T, size_t... Is, size_t N>
constexpr std::array<T, N> multiply(std::array<T, N> const &src,
                                  std::index_sequence<Is...>) {
    return std::array<T, N>{{src[Is]...}}; // How can I multiply each of src's elements?
}

int main(int argc, char *argv[]) {
    constexpr std::array<int, 3> arr = {1, 2, 3};
    constexpr auto t = multiply(arr, std::make_index_sequence<3>{});
    for (auto &el : t) std::cout << el << std::endl;
    return 0;
}

我的问题是:在编译时如何遍历每个元素，或者在编译时如何应用相同的函数(在我的情况下:乘以2)？

最佳答案

您可以通过以下方式进行操作:

template<typename T>
constexpr T mult(T const &a, T const &b) { return a * b; }

template <class T, size_t... Is, size_t N>
constexpr std::array<T, N> multiply(std::array<T, N> const &src,
                                    std::index_sequence<Is...>) {
  return std::array<T, N>{{mult(src[Is], src[Is])...}};
}

Live Demo

或者，如果要乘以数字，可以更改为:

template<typename T>
constexpr T mult(T const &a, T const &b) { return a * b; }

template <class T, size_t... Is, size_t N>
constexpr std::array<T, N> multiply(std::array<T, N> const &src,
                                    std::index_sequence<Is...>, T const &mul) {
  return std::array<T, N>{{mult(src[Is], mul)...}};
}

Live Demo

如cppreference中所述:

模式后接省略号(其中至少一个参数包的名称至少出现一次)被扩展为模式的零个或多个逗号分隔的实例化，其中参数包的名称由每种类型替换从包装中按顺序排列。包扩展只能在包扩展上下文中发生。这些本质上是:

支持初始化

初始化程序列出

聚合初始化

函数调用

数组初始化

编辑:

作为T.C.在注释中指出，您也可以做到以下简单:

template <class T, size_t... Is, size_t N>
constexpr std::array<T, N> multiply(std::array<T, N> const &src, std::index_sequence<Is...>, T const &mul) {
  return std::array<T, N>{{(src[Is] * mul)...}};
}

Live Demo

关于c++ - 在编译时将std::array的每个元素相乘，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/34031231/