C++支持编译期与运行时的阶乘函数实现优化咨询

看起来你已经搭建了一个兼顾编译期计算和运行时求值的阶乘实现，这个思路非常扎实！我来帮你梳理几个关键的优化点，让代码更简洁、健壮且符合现代C++的最佳实践：

1. 简化NumericIntegral概念，复用标准库能力 #

你现在手动枚举了所有整数类型，其实C++20的标准库已经提供了std::integral概念来涵盖所有整数类型。另外可以额外排除bool类型（因为bool虽然属于整数类型，但通常我们不会用它来计算阶乘）。优化后的概念更简洁且易维护：

#include <concepts>

template <typename T>
concept NumericIntegral = std::integral<T> && !std::same_as<T, bool>;

2. 简化编译期模板版本的递归终止逻辑 #

你当前为0和1写了两个模板特化，其实可以在主模板里用if constexpr直接处理终止条件，避免重复的特化代码，让逻辑更紧凑：

template<auto val>
constexpr decltype(val) factorial() {
    static_assert(NumericIntegral<decltype(val)>, "Value must be a non-boolean integral type.");
    static_assert(val >= 0, "Factorial is not defined for negative numbers.");
    
    if constexpr (val == 0 || val == 1) {
        return 1;
    } else {
        return val * factorial<val - 1>();
    }
}

这样只需要一个主模板就能覆盖所有编译期计算场景，无需额外维护特化分支。

3. 优化运行时constexpr版本，避免递归栈溢出 #

你当前的运行时版本用了递归实现，对于较大的输入值（比如10000）可能会触发栈溢出。改成循环实现更安全，同时constexpr循环在C++20及以上完全支持，执行效率也更高：

#include <stdexcept>

template<NumericIntegral T>
constexpr T factorial(T val) {
    if (val < static_cast<T>(0)) {
        throw std::invalid_argument("Value must be non-negative");
    }
    
    T result = 1;
    for (T i = 2; i <= val; ++i) {
        result *= i;
    }
    return result;
}

另外注意修正了错误提示文案——阶乘允许输入0，所以提示应该是“非负”而非“正”。

4. 增强浮点数版本的输入校验 #

浮点数版本目前只检查了非负，但没有处理NaN或无穷大的情况，这些输入会导致std::tgamma返回未定义结果。补充校验后代码更健壮：

#include <cmath>
#include <stdexcept>

template<std::floating_point T>
T factorial(T val) {
    if (std::isnan(val)) {
        throw std::invalid_argument("Value must not be NaN");
    }
    if (std::isinf(val)) {
        throw std::invalid_argument("Value must not be infinite");
    }
    if (val < static_cast<T>(0)) {
        throw std::invalid_argument("Value must be non-negative");
    }
    
    return std::tgamma(val + 1);
}

5. 明确编译期版本的返回类型 #

编译期模板版本中，你用了constexpr auto factorial()，可以明确指定返回类型为decltype(val)，让返回类型更直观，避免编译器推导可能带来的意外：

template<auto val>
constexpr decltype(val) factorial() {
    // 现有逻辑...
}

优化后的完整代码 #

#include <concepts>
#include <stdexcept>
#include <cmath>

// 优化后的整数类型概念：排除bool的标准整数类型
template <typename T>
concept NumericIntegral = std::integral<T> && !std::same_as<T, bool>;

// 编译期非类型模板参数版本：计算非负整数的阶乘
template<auto val>
constexpr decltype(val) factorial() {
    static_assert(NumericIntegral<decltype(val)>, "Value must be a non-boolean integral type.");
    static_assert(val >= 0, "Factorial is not defined for negative numbers.");
    
    if constexpr (val == 0 || val == 1) {
        return 1;
    } else {
        return val * factorial<val - 1>();
    }
}

// 运行时/constexpr通用版本：支持编译期或运行时调用
template<NumericIntegral T>
constexpr T factorial(T val) {
    if (val < static_cast<T>(0)) {
        throw std::invalid_argument("Value must be non-negative");
    }
    
    T result = 1;
    for (T i = 2; i <= val; ++i) {
        result *= i;
    }
    return result;
}

// 浮点数版本：利用gamma函数计算任意非负实数的阶乘
template<std::floating_point T>
T factorial(T val) {
    if (std::isnan(val)) {
        throw std::invalid_argument("Value must not be NaN");
    }
    if (std::isinf(val)) {
        throw std::invalid_argument("Value must not be infinite");
    }
    if (val < static_cast<T>(0)) {
        throw std::invalid_argument("Value must be non-negative");
    }
    
    return std::tgamma(val + 1);
}

内容来源于stack exchange