掌握C++17中的折叠表达式：从基础到实战

作者

折叠表达式（Fold Expression）是 C++17 引入的一项强大功能，它可以让我们用极简的语法完成对可变参数模板（Variadic Templates）中参数的聚合操作。无论是求和、相乘、按位与或或，甚至更复杂的组合逻辑，都能通过一行代码实现。本文将从折叠表达式的基本语法讲起，逐步演示其常见用法，并给出若干实战案例，帮助你在日常项目中快速上手。

1. 折叠表达式的语法结构

折叠表达式分为三类：

结构 说明 例子
( init op ... ) 左折叠（从左到右） (0 + ...) 计算求和，等价于 0 + a1 + a2 + ...
(... op init) 右折叠（从右到左） (... * 1) 计算乘积，等价于 a1 * a2 * ... * 1
(... op ...) 全折叠（左右折叠，先左后右） (... + ...) 计算求和，等价于 a1 + a2 + ...

需要注意的是，折叠表达式只能作用于可变参数包（Args...），并且其运算符必须是二元运算符。折叠表达式的左右折叠顺序决定了运算的优先级，尤其在不具备结合性的运算符（如除法、取模等）时要特别小心。

2. 基础实例

2.1 求和

template <typename... Args>
constexpr auto sum(Args&&... args) {
    return (0 + ... + std::forward <Args>(args));
}

调用 sum(1, 2, 3, 4) 将返回 10。

2.2 乘积

template <typename... Args>
constexpr auto product(Args&&... args) {
    return (... * std::forward <Args>(args));
}

调用 product(2, 3, 4) 将返回 24。

2.3 判断所有元素是否满足条件

template <typename Predicate, typename... Args>
constexpr bool all_of(Predicate pred, Args&&... args) {
    return (... && pred(std::forward <Args>(args)));
}

使用方式 all_of([](int x){ return x > 0; }, 1, 2, 3) 返回 true。

3. 进阶用法

3.1 与 std::initializer_list 结合

在 C++17 之前，常见的做法是使用 std::initializer_list 来进行求和：

int sum = { args... } + 0;  // 需要自定义加法运算符

折叠表达式使得上述操作更加直观，避免了显式的 std::initializer_list。

3.2 与 constexpr 结合

折叠表达式可以在编译期执行，这使得我们可以在 constexpr 函数中完成复杂的计算。

constexpr int factorial(int n) {
    return (n == 0) ? 1 : (n * factorial(n - 1));
}

但若需要对多值进行折叠，仍需手动写模板。

3.3 自定义运算符

折叠表达式可以使用任何自定义的二元运算符，只要它在模板参数中可用。例如：

struct And {
    bool operator()(bool a, bool b) const { return a && b; }
};

template <typename... Args>
constexpr bool all_true(Args&&... args) {
    return (And{}(..., std::forward <Args>(args)));
}

4. 实战案例

4.1 变参日志系统

我们经常需要实现一个可变参数日志函数，支持任意数量的参数，且参数可以是不同类型。折叠表达式可以帮助我们轻松实现。

#include <iostream>
#include <string>

void log(const std::string& prefix) {
    std::cout << prefix << std::endl;
}

template <typename T, typename... Args>
void log(const std::string& prefix, const T& first, const Args&... rest) {
    std::cout << prefix << " " << first;
    if constexpr (sizeof...(rest) > 0) {
        std::cout << " | ";
        log("", rest...);
    } else {
        std::cout << std::endl;
    }
}

调用 log("INFO", "用户", 12345, "已登录"); 会输出：

INFO 用户 | 12345 | 已登录

折叠表达式也可以用来一次性处理所有参数：

template <typename... Args>
void log2(const std::string& prefix, const Args&... args) {
    ((std::cout << prefix << " " << args << " | "), ...);
    std::cout << std::endl;
}

4.2 可变模板参数的多态

假设你需要实现一个通用的 apply 函数，它接受一个函数对象和一组参数，使用折叠表达式将所有参数一次性传递给函数：

template <typename F, typename... Args>
auto apply(F&& f, Args&&... args) {
    return std::forward <F>(f)(std::forward<Args>(args)...);
}

使用 apply 时无需担心参数数量，只需关注函数签名即可。

4.3 递归与折叠的结合

有时我们需要在折叠表达式中做递归操作，例如实现一个“链式调用”的 DSL。下面示例展示如何用折叠表达式为链式调用添加日志：

struct Chain {
    void step(int value) {
        std::cout << "Step: " << value << std::endl;
    }
};

template <typename... Args>
void chain(Chain& c, Args&&... args) {
    (c.step(args), ...);
}

调用 chain(chainObj, 1, 2, 3); 将依次输出三步。

5. 常见陷阱与最佳实践

  1. 左折叠与右折叠的区别

    • 对于 结合性 的运算符（如 +, *），左折叠和右折叠得到相同结果。
    • 对于 不结合性 的运算符（如 /, %），左折叠与右折叠的顺序会影响结果。
    • 习惯使用 (... op ...) 的全折叠来避免手动指定方向。

  2. 默认值的选择

    • 在折叠表达式中提供一个合适的初始值（如 、1、true、false）可以使表达式更直观。
    • 如果没有初始值，则使用全折叠 ( ... op ...)。

  3. 可变参数数量为零时

    • 全折叠 ( ... op ... ) 在参数数量为零时会导致编译错误。
    • 可通过 if constexpr 或 sizeof...(args) 检查来处理空参数包。

  4. 避免滥用

    • 虽然折叠表达式能让代码简洁，但在可读性更重要的场景下，还是建议保持传统循环或递归实现。

6. 小结

折叠表达式让我们可以以极简的方式处理可变参数模板，为 C++17 的可变参数编程打开了新维度。掌握其基本语法、结合使用以及常见陷阱后，你就能在日常项目中快速实现求和、乘积、布尔聚合以及自定义运算等功能。随着模板元编程和 constexpr 的进一步发展，折叠表达式必将成为 C++ 高效代码不可或缺的一部分。祝你编码愉快，折叠无极限！

发表评论