跳转至

Lambda 表达式

注意:考虑到算法竞赛的实际情况,本文将不会全面研究语法,只会讲述在算法竞赛中可能会应用到的部分。

本文语法参照 C++11 标准。语义不同的将以 C++11 作为标准,C++14、C++17 的语法视情况提及并会特别标注。

Lambda 表达式

Lambda 表达式因数学中的 演算得名,直接对应于其中的 lambda 抽象。Lambda 表达式能够捕获作用域中的变量的无名函数对象。我们可以将其理解为一个匿名的内联函数,可以用来替换独立函数或者函数对象,从而使代码更可读。但是从本质上来讲,Lambda 表达式只是一种语法糖,因为它能完成的工作也可以用其他复杂的 C++ 语法来实现。

下面是 Lambda 表达式的语法:

1
[capture] (parameters) mutable -> return-type {statement}

下面我们分别对其中的 capture, parameters, mutable, return-type, statement 进行介绍。

capture 捕获子句

Lambda 表达式以 capture 子句开头,它指定哪些变量被捕获,以及捕获是通过值还是引用:有 & 符号前缀的变量通过引用访问,没有该前缀的变量通过值访问。空的 capture 子句 [] 指示 Lambda 表达式的主体不访问封闭范围中的变量。

我们也可以使用默认捕获模式:& 表示捕获到的所有变量都通过引用访问,= 表示捕获到的所有变量都通过值访问。之后我们可以为特定的变量 显式 指定相反的模式。

例如 Lambda 体要通过引用访问外部变量 a 并通过值访问外部变量 b,则以下子句等效:

  • [&a, b]
  • [b, &a]
  • [&, b]
  • [b, &]
  • [=, &a]

默认捕获时,会捕获 Lambda 中提及的变量。获的变量成为 Lambda 的一部分;与函数参数相比,调用 Lambda 时不必传递它们。

以下是一些常见的例子:

1
2
3
4
5
6
int a = 0;
auto f = []() { return a * 9; };   // Error, 无法访问 'a'
auto f = [a]() { return a * 9; };  // OK, 'a' 被值「捕获」
auto f = [&a]() { return a++; };   // OK, 'a' 被引用「捕获」
                                  // 注意:请保证 Lambda 被调用时 a 没有被销毁
auto b = f();  // f 从捕获列表里获得 a 的值,无需通过参数传入 a

parameters 参数列表

大多数情况下类似于函数的参数列表,例如:

1
2
auto lam = [](int a, int b) { return a + b; };
std::cout << lam(1, 9) << " " << lam(2, 6) << std::endl;

C++14 中,若参数类型是泛型,则可以使用 auto 声明类型:

1
auto lam = [](auto a, auto b)

一个例子:

1
2
3
int x[] = {5, 1, 7, 6, 1, 4, 2};
std::sort(x, x + 7, [](int a, int b) { return (a > b); });
for (auto i : x) std::cout << i << " ";

这将打印出 x 数组从大到小排序后的结果。

由于 parameters 参数列表 是可选的,如果不将参数传递给 Lambda 表达式,并且其 Lambda 声明器不包含 mutable,且没有后置返回值类型,则可以省略空括号。

Lambda 表达式也可以将另一个 Lambda 表达式作为其参数。

一个例子:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
#include <functional>
#include <iostream>

int main() {
  using namespace std;

  // 返回另一个计算两数之和 Lambda 表达式
  auto addtwointegers = [](int x) -> function<int(int)> {
    return [=](int y) { return x + y; };
  };

  // 接受另外一个函数 f 作为参数,返回 f(z) 的两倍
  auto higherorder = [](const function<int(int)>& f, int z) {
    return f(z) * 2;
  };

  // 调用绑定到 higherorder 的 Lambda 表达式
  auto answer = higherorder(addtwointegers(7), 8);

  // 答案为 (7 + 8) * 2 = 15
  cout << answer << endl;
}

mutable 可变规范

利用可变规范,Lambda 表达式的主体可以修改通过值捕获的变量。若使用此关键字,则 parameters 不可省略(即使为空)。

一个例子,使用 capture 捕获字句 中的例子,来观察 a 的值的变化:

1
2
int a = 0;
auto func = [a]() mutable { ++a; };

此时 lambda 中的 a 的值改变为 1,lambda 外的 a 保持不变。

return-type 返回类型

用于指定 Lambda 表达式的返回类型。若没有指定返回类型,则返回类型将被自动推断(行为与用 auto 声明返回值的普通函数一致)。具体的,如果函数体中没有 return 语句,返回类型将被推导为 void,否则根据返回值推导。若有多个 return 语句且返回值类型不同,将产生编译错误。

例如,上文的 lam 也可以写作:

1
auto lam = [](int a, int b) -> int

再举两个例子:

1
2
auto x1 = [](int i) { return i; };  // OK
auto x2 = [] { return {1, 2}; };    // Error, 返回类型被推导为 void

statement Lambda 主体

Lambda 主体可包含任何函数可包含的部分。普通函数和 Lambda 表达式主体均可访问以下变量类型:

  • 从封闭范围捕获变量
  • 参数
  • 本地声明的变量
  • 在一个 class 中声明时,若捕获 this,则可以访问该对象的成员
  • 具有静态存储时间的任何变量,如全局变量

下面是一个例子

1
2
3
4
5
6
7
8
#include <iostream>

int main() {
  int m = 0, n = 0;
  [&, n](int a) mutable { m = (++n) + a; }(4);
  std::cout << m << " " << n << std::endl;
  return 0;
}

最后我们得到输出 5 0。这是由于 n 是通过值捕获的,在调用 Lambda 表达式后仍保持原来的值 0 不变。mutable 规范允许 n 在 Lambda 主体中被修改,将 mutable 删去则编译不通过。

使用类完成更复杂的操作

在 C++11 前没有 Lambda 表达式,但可以使用稍复杂的方法替代,尽管看上去更复杂却更易理解及扩展。

首先我们已经知道 Lambda 本质是一个可调用的对象,那么直接定义一个类并构造一个对象,重载其 operator() 运算符就可以完成和 Lambda 一样的操作,下面看一个简单的例子,我们将使用 C++17 的语法:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
#include <iostream>

struct AbstractCallable {
  AbstractCallable(int v) : v_(v) {}

  virtual ~AbstractCallable() = default;
  virtual int operator()(int) const = 0;  // 纯虚函数需要继承后实现才可以实例化
  int v_;
};

AbstractCallable *get_CallableObject() {
  struct Callable : public AbstractCallable {
    Callable() : AbstractCallable(10) {}

    int operator()(int k) const override {
      std::cout << v_ + k << std::endl;
      return v_ - 10;
    }
  };

  return new Callable;
}

int main() {
  auto t = get_CallableObject();
  std::cout << t->operator()(5);  // 或者等价的 `(*t)(5);`
  delete t;
  return 0;
}

在写 Lambda 表达式时,我们几乎都可以将其等价的映射为上面这种形式。

Lambda 表达式相关语法类的语法
capture 捕获子句构造函数
-析构函数
使用 std::function 包装传递基类指针/引用传递
拷贝多个 Lambda 的函数对象自定义的拷贝函数
mutableoperator() 函数是否为 const

在 Lambda 的捕获子句中分为引用捕获和按值捕获(暂不考虑比较特殊的捕获 this 等),而在类的构造函数中我们可以更精细的控制这一点,另外自定义的析构函数的存在也方便我们更好的扩展,缺点是不够「匿名」,因为仍需要类名。

假设我们有一个函数

1
void func(std::function<int(int)>);

那么上述用例中就可以改为

1
void func(AbstractCallable *);

并且既然 Callable 是一个可调用对象,我们也可以通过 std::bind(&AbstractCallable::operator(), t, std::placeholders::_1) 来将其转换再转换为 std::function<int(int)>

如果不需要实现类似 std::function 的包装,那么也无需使用抽象基类,这样便和一般的 Lambda 表达式一样不会产生额外的虚拟函数表的开销。

参考文献