如何正确返回模板lambda函数实现C++多态柯里化take函数？

你遇到的编译报错是因为模板参数T仅出现在函数返回值类型中，调用take(2)时没有任何传入参数可以为编译器提供T的类型推导依据，因此模板匹配失败。

最简单的修复方式是调用take时显式声明元素类型，可直接复用你现有的模板实现：

// 调用时显式指定T为int
std::function<std::vector<int>(std::vector<int>)> take_2 = take<int>(2);

该方案的缺点是每次部分应用take时都需要提前指定容器元素类型，灵活度较低。

如果你的编译环境支持C++14及以上标准，可以直接返回带auto参数的泛型lambda，完全避免提前声明模板参数的问题，更接近Haskell的多态函数行为：

#include <vector>
#include <algorithm>

// C++14 兼容实现，支持任意带有resize、size方法和value_type定义的容器
auto take(int n) {
  return [=](auto container) {
    // 取n和容器长度的最小值，和Haskell take行为保持一致，避免resize补默认构造元素
    const auto take_cnt = std::min<std::size_t>(n, container.size());
    container.resize(take_cnt);
    return container;
  };
}

如果是C++20及以上环境，还可以通过显式lambda模板参数限制输入为std::vector，类型约束更严格：

// C++20 版本，仅支持std::vector输入
auto take(int n) {
  return [=]<typename T>(std::vector<T> v) {
    const auto take_cnt = std::min<std::size_t>(n, v.size());
    v.resize(take_cnt);
    return v;
  };
}

使用示例 #

// 部分应用，不需要提前指定元素类型，支持任意符合要求的容器输入
auto take_2 = take(2);

// 测试int类型vector
std::vector<int> int_vec = {1,2,3,4,5};
std::vector<int> res_int = take_2(int_vec); // 得到 {1,2}

// 测试string类型vector
std::vector<std::string> str_vec = {"a","b","c","d"};
std::vector<std::string> res_str = take_2(str_vec); // 得到 {"a","b"}

方案优势 #

• 完全实现了类似Haskell的参数多态特性，一次部分应用得到的take_n函数可以处理任意元素类型的容器
• 不需要依赖std::function的包装，运行时开销更低
• 调用时不需要手动指定模板参数，语法更简洁

如果你需要兼容C++11标准，无法使用泛型lambda，可以将取元素的逻辑封装为模板调用类，或者继续使用原来的模板函数+显式指定类型的方案。

内容的提问来源于stack exchange，提问作者Finley