如何为指针重载[]运算符？C++结构体枚举场景问题

核心问题分析 #

你遇到的两个问题本质是：

1. 返回类型不统一：_id（int）和_name（std::string）属于不同类型，普通运算符重载无法同时返回两种类型；
2. 原生指针无法重载运算符：C++不允许修改原生指针的内置operator[]语义（原生指针pt[n]等价于*(pt+n)，用于数组元素访问），必须通过包装类实现自定义逻辑。

方案1：代理对象法（贴近需求语法） #

通过代理类的类型转换运算符，适配不同成员的类型，实现和pt[ID]/pt[NAME]一致的调用语法。

完整代码 #

#include <string>
#include <stdexcept>

struct T{
    int _id;
    std::string _name;
};

enum {ID = 1, NAME};

// 代理类：适配不同类型的成员访问
struct MemberProxy {
    T* ptr;
    int member_type;

    // 转换为int&，用于访问_id
    operator int&() {
        if (member_type == ID) {
            return ptr->_id;
        }
        throw std::bad_cast("错误：无法将ID成员转换为string类型");
    }

    // 转换为std::string&，用于访问_name
    operator std::string&() {
        if (member_type == NAME) {
            return ptr->_name;
        }
        throw std::bad_cast("错误：无法将NAME成员转换为int类型");
    }
};

// T*的包装类：实现自定义operator[]
struct TPtrWrapper {
    T* ptr;

    MemberProxy operator[](int type) {
        return {ptr, type};
    }
};

// 使用示例
int main() {
    T obj{100, "test"};
    TPtrWrapper pt{&obj};

    // 替代pt->_id = 200;
    pt[ID] = 200;
    // 替代int id = pt->_id;
    int id = pt[ID];

    // 替代pt->_name = "new_name";
    pt[NAME] = "new_name";
    // 替代std::string name = pt->_name;
    std::string name = pt[NAME];

    return 0;
}

方案2：C++17 std::variant法 #

如果使用C++17及以上标准，可通过std::variant返回多类型引用，需配合std::get获取对应类型。

完整代码 #

#include <string>
#include <variant>
#include <stdexcept>

struct T{
    int _id;
    std::string _name;
};

enum {ID = 1, NAME};

// T*的包装类
struct TPtrWrapper {
    T* ptr;

    std::variant<int&, std::string&> operator[](int type) {
        switch(type) {
            case ID: return ptr->_id;
            case NAME: return ptr->_name;
            default: throw std::invalid_argument("错误：无效的成员类型");
        }
    }
};

// 使用示例
int main() {
    T obj{100, "test"};
    TPtrWrapper pt{&obj};

    std::get<int&>(pt[ID]) = 200;
    int id = std::get<int&>(pt[ID]);

    std::get<std::string&>(pt[NAME]) = "new_name";
    std::string name = std::get<std::string&>(pt[NAME]);

    return 0;
}

注意事项 #

• 代理对象法的语法更贴合你的需求，但错误类型转换（如将pt[ID]赋值给std::string）会抛出异常，建议生产环境补充错误处理逻辑；
• 包装类是必须的，因为C++不允许重载原生指针的运算符，无法修改其内置数组访问语义。

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