C++ lambda 返回自己

C++ lambda returning itself

本文关键字:自己 返回 lambda C++      更新时间:2023-10-16

我想写一个返回自身的lambda,这样我就可以当场多次调用它。但是看起来在lambda内部this指的是lambda,而是指周围对象的this,如果lambda是在成员函数中定义的。

下面是一个示例:

#include <iostream>
int main(int argc, char* argv[]) {
int a = 5;
[&](int b) {
std::cout << (a + b) << std::endl;
return *this;
}(4)(6);
}

有没有办法做一些类似的事情?

使用旧函子:

int main() {
int a = 5;
struct S {
const S& operator ()(int b) const {
std::cout << (a + b) << std::endl;
return *this;
}
const int& a;
};
S{a}(4)(6);
}

演示

你不能返回 lambda 本身,但你可以返回一个不同的:

#include <iostream>
int main() {
int a = 5;    
[&](int b) {
auto impl = [&](int b){std::cout << (a + b) << std::endl;};
impl(b);
return impl;
}(4)(6);
}

但是,这只允许再调用一次。不确定是否有一些技巧可以从中获得更多...

Ben Voigt 建议使用 Y 组合子(这对标准库 BTW 来说是一个很好的建议(,但你的问题更简单。您可以引入一个小的模板函子来代替 lambda:

template<typename T>
struct recallable_impl {
template<typename... Args>
recallable_impl& operator()(Args&&... args) {
f(std::forward<Args>(args)...);
return *this;
}
template<typename F>
recallable_impl(F&& f)
:  f{std::forward<F>(f)}
{}
private:
T f;
};
template<typename T>
decltype(auto) recallable(T&& f) {
return recallable_impl<std::decay_t<T>>(std::forward<T>(f));
}

然后,您的 lambda 甚至不需要显式返回任何内容:

int main() {
int a = 5;  
recallable([&](int b){std::cout << (a + b) << std::endl;})(4)(5)(6)(7)(8);
}

如果您将lambda 分配给预置类型(不是 auto(,则可以调用它,并且它会被捕获:

std::function<void(int)> f =[&f](int n)
{
if (n>0) f(n-1);
return;
};