如何告知编译器模板参数U等价于std::vector<T>或T？

给定模板函数定义：

template<typename T, typename U> std::vector<T> foo(T t, U){return std::vector<T>{};}

要让编译器仅允许模板参数U为std::vector<T>或T类型，可通过以下几种方式实现：

方案1：C++20 概念（Concepts） #

C++20引入的概念机制能直观定义模板参数约束，代码可读性更高：

#include <vector>
#include <concepts>

// 定义约束U的概念
template<typename T, typename U>
concept ValidU = std::same_as<U, T> || std::same_as<U, std::vector<T>>;

// 用概念约束U的类型
template<typename T, ValidU<T> U>
std::vector<T> foo(T t, U) {
    return std::vector<T>{};
}

当U不符合约束时，编译器会直接拒绝模板实例化，报错信息清晰。

方案2：C++20 之前的SFINAE方法 #

针对C11/C17等旧标准，可借助SFINAE特性结合类型萃取实现：

#include <vector>
#include <type_traits>

// 仅当U是T或std::vector<T>时，才会生成该函数实例
template<typename T, typename U>
std::enable_if_t<std::is_same_v<U, T> || std::is_same_v<U, std::vector<T>>, std::vector<T>>
foo(T t, U) {
    return std::vector<T>{};
}

std::enable_if_t会在条件满足时才启用当前模板，否则替换失败，编译器不会选择该模板。

方案3：静态断言（Static Assert） #

如果不需要处理模板重载场景，静态断言是更简单的选择，能自定义报错信息：

#include <vector>
#include <type_traits>

template<typename T, typename U>
std::vector<T> foo(T t, U) {
    // 实例化时检查类型约束，不满足则触发编译错误
    static_assert(std::is_same_v<U, T> || std::is_same_v<U, std::vector<T>>,
                  "模板参数U必须是T类型或std::vector<T>类型");
    return std::vector<T>{};
}

这种方式在模板被实例化后才会检查条件，报错信息更直白，适合简单场景。

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