如何实现接受指定尺寸std::array的函数并支持模板推导？

1. 能否让PickyArrayHandler自动推导数组尺寸？ #

当然可以！原来的代码无法自动推导，核心原因是模板参数用了枚举类型（SizesForArrayX SX），而std::array的尺寸是size_t类型——编译器没办法从数组的size_t尺寸反向推导出对应的枚举值。我们只需要调整模板参数的设计，直接用数组的尺寸作为模板参数，再配合编译时检查来验证尺寸是否合法，就能实现自动推导。

2. 更优的实现方案 #

最优方案的核心是分离“允许的尺寸集合”和“函数逻辑”，这样未来尺寸集合变更时，只需要修改集合定义，函数本身完全不用动。同时利用C++14的constexpr特性做编译时合法性检查，既保证类型安全，又不影响运行时性能。

具体实现代码 #

#include <type_traits>
#include <array>

// 第一步：定义允许的尺寸集合（未来变更只需要修改这里）
// 可以用枚举转成constexpr数组，或者直接写数值
enum class SizesForArrayX : size_t { Three = 3, Four, Forty = 40 };
enum class SizesForArrayY : size_t { Two = 2, Three, EleventyTwelve = 122 };

// 把枚举转成constexpr数组，方便后续检查
constexpr std::array<size_t, 3> allowed_sizes_x = {
    static_cast<size_t>(SizesForArrayX::Three),
    static_cast<size_t>(SizesForArrayX::Four),
    static_cast<size_t>(SizesForArrayX::Forty)
};

constexpr std::array<size_t, 3> allowed_sizes_y = {
    static_cast<size_t>(SizesForArrayY::Two),
    static_cast<size_t>(SizesForArrayY::Three),
    static_cast<size_t>(SizesForArrayY::EleventyTwelve)
};

// 第二步：编译时检查工具函数——判断尺寸是否在允许集合中
template <size_t N>
constexpr bool is_size_allowed(size_t size, const std::array<size_t, N>& allowed_set) {
    for (size_t i = 0; i < N; ++i) {
        if (allowed_set[i] == size) {
            return true;
        }
    }
    return false;
}

// 第三步：最终的函数模板——自动推导尺寸+编译时检查
template <size_t SX, size_t SY>
void PickyArrayHandler(const std::array<int, SX>& x, const std::array<int, SY>& y) {
    // 编译时断言，非法尺寸直接报错
    static_assert(is_size_allowed(SX, allowed_sizes_x), 
                  "Error: X array size is not in the allowed set!");
    static_assert(is_size_allowed(SY, allowed_sizes_y), 
                  "Error: Y array size is not in the allowed set!");

    // 原函数逻辑保持不变
    for (auto& i : x) i = 42;
    for (auto& i : y) while (i --> -41) i = i;
}

// 测试用例
int main() {
    std::array<int, 3> x_legal;
    std::array<int, 2> y_legal;
    PickyArrayHandler(x_legal, y_legal); // 正常编译，自动推导SX=3，SY=2

    // std::array<int, 5> x_illegal;
    // PickyArrayHandler(x_illegal, y_legal); // 编译报错，提示尺寸非法
    return 0;
}

方案优势 #

• 自动推导：调用时直接传入std::array对象，编译器会自动推导模板参数SX和SY，无需显式指定。
• 低耦合：允许的尺寸集合集中定义，未来新增/修改尺寸时，只需要更新allowed_sizes_x或allowed_sizes_y，函数逻辑完全不用改动。
• 编译时安全：通过static_assert在编译阶段就拦截非法尺寸，避免运行时错误。
• 简洁可扩展：即使允许的尺寸数量很多，只需要在constexpr数组里添加元素即可，比函数重载或大量模板特化高效得多。

内容的提问来源于stack exchange，提问作者quasinormalized