在C++中对复数集进行自定义[]访问

#include <iostream>
#include <vector>

class Complex {
private:
    double real;
    double imag;

public:
    Complex(double r, double i) : real(r), imag(i) {}

    double getReal() const {
        return real;
    }

    double getImag() const {
        return imag;
    }
};

class ComplexSet {
private:
    std::vector<Complex> data;

public:
    ComplexSet() {}

    ComplexSet(const std::vector<Complex>& d) : data(d) {}

    Complex& operator[](int index) {
        return data[index];
    }

    const Complex& operator[](int index) const {
        return data[index];
    }
};

int main() {
    std::vector<Complex> complexData;
    complexData.push_back(Complex(1.0, 2.0));
    complexData.push_back(Complex(3.0, 4.0));
    complexData.push_back(Complex(5.0, 6.0));

    ComplexSet complexSet(complexData);

    std::cout << "Real part of complex number at index 1: " << complexSet[1].getReal() << std::endl;
    std::cout << "Imaginary part of complex number at index 2: " << complexSet[2].getImag() << std::endl;

    return 0;
}