如何让自定义类支持通过索引访问std::vector<std::variant>元素？

你的原operator[]实现存在两个核心问题：

1. std::visit的lambda返回的是临时复制的对象（static_cast<T>(value)），无法返回引用，会导致编译错误。
2. 未处理variant实际类型与模板参数T不匹配的情况，强行转换会引发未定义行为。

下面提供两种可行的实现方式：

方案一：直接使用std::get获取类型引用 #

利用std::variant的std::get<T>接口直接提取对应类型的引用，实现最简单直接：

struct Data {
    using data_types = std::variant<std::basic_string<char>, double, int>;

public:
    template <class T>
    void push_back(const T& t) {
        m_data.push_back(t);
    }

    // 非const版本，返回可修改引用
    template <class T>
    T& operator[](size_t index) {
        return std::get<T>(m_data[index]);
    }

    // const版本，返回只读引用
    template <class T>
    const T& operator[](size_t index) const {
        return std::get<T>(m_data[index]);
    }

private:
    std::vector<data_types> m_data;
};

使用方式 #

由于模板参数T无法自动推导，需要显式指定类型：

int main() {
    Data d;
    d.push_back(0);
    d.push_back("string");
    d.push_back(3.55);

    int& i = d.operator[]<int>(0);
    std::string& s = d.operator[]<std::string>(1);
    double& dbl = d.operator[]<double>(2);

    // 修改值
    i = 10;
    s = "new string";
    dbl = 4.66;
}

如果类型不匹配（比如用std::string访问索引0），会直接抛出std::bad_variant_access异常。

方案二：用std::visit配合if constexpr实现类型安全访问 #

通过if constexpr在编译期判断variant的实际类型，确保只有类型匹配时才返回引用，否则抛出异常：

#include <stdexcept> // 用于std::bad_variant_access

struct Data {
    using data_types = std::variant<std::basic_string<char>, double, int>;

public:
    template <class T>
    void push_back(const T& t) {
        m_data.push_back(t);
    }

    template <class T>
    T& operator[](size_t index) {
        return std::visit([](auto&& value) -> T& {
            using ValueType = std::decay_t<decltype(value)>;
            if constexpr (std::is_same_v<ValueType, T>) {
                return value;
            } else {
                throw std::bad_variant_access{};
            }
        }, m_data[index]);
    }

    template <class T>
    const T& operator[](size_t index) const {
        return std::visit([](auto&& value) -> const T& {
            using ValueType = std::decay_t<decltype(value)>;
            if constexpr (std::is_same_v<ValueType, T>) {
                return value;
            } else {
                throw std::bad_variant_access{};
            }
        }, m_data[index]);
    }

private:
    std::vector<data_types> m_data;
};

使用方式 #

和方案一完全一致，显式指定模板参数即可。这种方式的优势是逻辑更清晰，能明确处理类型不匹配的场景。

注意事项 #

• 两种方案都要求你明确知道每个索引对应的variant类型，否则会触发异常。
• 必须提供const版本的operator[]，否则无法在const Data对象上使用该操作。
• 需要C17及以上标准支持（std::variant、if constexpr都是C17特性）。

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