在 cpp 中构建一个通用的 `map` 类函数

Question

考虑

template <typename S, typename T, typename F>
vector<T> map(const vector<S> &ss, F f){
    vector<T> ts;
    ts.reserve(ss.size());
    std::transform(ss.begin(), ss.end(), std::back_inserter(ts), f);
    return ts;
}

int main(){
    vector<int> is = {...};
    vector<double> ts = map(is, [](int i){return 1.2*i;});
}

因为编译器“无法推导出模板参数 T”。指定类型

的 map

template <typename S, typename T>
vector<T> map(const vector<S> &ss, std::function<T(S)> f)

也不行，因为它不匹配 lambda。

正确的方法是什么？

Answer 1

像这样：

template <typename S, typename F>
auto map(const std::vector<S> &ss, F f) -> std::vector<std::decay_t<decltype(f(ss[0]))>>
{
    std::vector<std::decay_t<decltype(f(ss[0]))>> ts;
    ts.reserve(ss.size());
    std::transform(ss.begin(), ss.end(), std::back_inserter(ts), std::move(f));
    return ts;
}

我选择了尾随 return 类型，以便能够在 decltype 中使用 f 和 ss。您可以只使用 auto return 类型，但明确指定类型会使函数 SFINAE-friendly.

或者，具有最热门 C++20 功能的稍微过度设计的版本：

template <
    template <typename...> typename C = std::vector,
    typename S, typename F
>
[[nodiscard]] C<std::decay_t<std::indirect_result_t<F, std::ranges::iterator_t<S>>>>
map(S &&ss, F f)
{
    C<std::decay_t<std::indirect_result_t<F, std::ranges::iterator_t<S>>>> ts;
    ts.reserve(std::ranges::size(ss));
    std::ranges::transform(ss, std::back_inserter(ts), std::move(f));
    return ts;
}

Answer 2

这有效

  std::vector<double> ts = map<int, double>(is, [](int i) {return 1.2 * i; });

在 cpp 中构建一个通用的 `map` 类函数

Build a generic `map` like function in cpp

c++

templates