C++17中的折叠表达式（Fold Expressions）详解

作者

在C++17之前，若需对可变参数模板（variadic template）进行归约（如求和、乘积、逻辑与/或等），开发者往往需要自行实现递归结构或利用初始化列表展开，代码相对冗长且易出错。C++17引入了折叠表达式（Fold Expressions），极大简化了对可变参数的处理，让模板编程更简洁、高效。本文将从概念、语法、常见用法以及注意事项四个方面，系统阐述折叠表达式的核心内容。

1. 折叠表达式概念

折叠表达式是将一个二元运算符（如 +, *, &&, ||, ==, 等）递归地应用于可变参数包（parameter pack）中的每个元素，最终得到一个单一的值。它的核心思想是“折叠”整个参数包为一个聚合结果。

1.1 语法结构

折叠表达式有四种基本形式：

  1. 左折叠（left fold） 

    ( pack op ... )   // 例： (a + b + c)

    等价于 (((a op b) op c) ...)

  2. 右折叠（right fold） 

    (... op pack)   // 例： (a + b + c)

    等价于 (a op (b op (c op ...)))

  3. 无运算符左折叠（unary left fold） 

    ( op ... pack )   // 例： (!a || !b || !c)

    适用于单目运算符。

  4. 无运算符右折叠（unary right fold） 

    (... op pack)   // 例： (a || b || c)

需要注意的是，在无运算符折叠中，运算符在包前还是包后决定折叠方向。

2. 常见用法举例

2.1 求和与乘积

template<typename... Args>
auto sum(Args&&... args) {
    return (args + ...);    // 右折叠
}

template<typename... Args>
auto product(Args&&... args) {
    return (args * ...);    // 右折叠
}

使用方式：

int main() {
    std::cout << sum(1, 2, 3, 4) << '\n';   // 输出 10
    std::cout << product(2, 3, 4) << '\n';  // 输出 24
}

2.2 逻辑与/或

template<typename... Bools>
bool all_true(Bools&&... bs) {
    return (bs && ...);    // 右折叠
}

template<typename... Bools>
bool any_true(Bools&&... bs) {
    return (bs || ...);    // 右折叠
}

2.3 元组展平（Flatten a tuple）

template<typename Tuple, std::size_t... I>
auto flatten_impl(Tuple&& t, std::index_sequence<I...>) {
    return std::make_tuple(std::get <I>(std::forward<Tuple>(t))...);
}

template<typename... Ts>
auto flatten(std::tuple<Ts...>&& t) {
    return flatten_impl(std::move(t), std::make_index_sequence<sizeof...(Ts)>{});
}

此处并未直接用折叠表达式，但通过折叠表达式可进一步简化：

template<typename Tuple, std::size_t... I>
auto flatten_impl(Tuple&& t, std::index_sequence<I...>) {
    return std::tuple_cat(std::forward <Tuple>(t)...); // 这并非折叠，需要额外逻辑
}

2.4 通过折叠实现可变参数的打印

template<typename... Args>
void print(const Args&... args) {
    ((std::cout << args << ' '), ...);  // 无运算符左折叠
    std::cout << '\n';
}

3. 关键注意点

  1. 运算符优先级
    折叠表达式需要与其他表达式分开，最好使用括号包围，以防优先级混淆。

  2. 空包的折叠
    对空参数包使用折叠表达式会导致编译错误。需要提供默认值： 

    template<typename... Args>
int sum(int init = 0, Args&&... args) {
    return (init + ... + args);
}

  3. 副作用与顺序
    折叠表达式的展开顺序（左折叠或右折叠）会影响副作用的执行顺序。若运算符具有副作用（如函数调用、递增操作），需谨慎使用。

  4. 递归模板简化
    通过折叠表达式，原本需要递归实现的逻辑可以在一行中完成，减少模板层级。

4. 实战案例：构造一个简单的日志系统

#include <iostream>
#include <string>
#include <chrono>
#include <ctime>

inline std::string now() {
    std::time_t t = std::chrono::system_clock::to_time_t(std::chrono::system_clock::now());
    return std::string(std::ctime(&t));
}

template<typename... Args>
void log(const Args&... args) {
    std::cout << "[" << now() << "] ";
    ((std::cout << args), ...);
    std::cout << '\n';
}

int main() {
    int user_id = 42;
    double balance = 1234.56;
    log("User ID: ", user_id, " | Balance: $", balance);
}

此处，折叠表达式 ( (std::cout << args), ... ) 实现了参数的逐个输出，且保持了输出顺序。

5. 结语

折叠表达式是 C++17 对模板元编程的一次重要补充，它让处理可变参数变得更加自然、简洁。掌握折叠表达式后，许多原本繁琐的递归模板实现可以被轻松替换为一行表达式，代码可读性和可维护性大幅提升。建议在日常项目中多尝试折叠表达式，尤其是数值归约、逻辑判断以及字符串拼接等场景，能显著提高代码质量与开发效率。

发表评论