重载向量的多维数组运算符

#include <iostream>
#include <vector>

class MultiDimensionalArray {
private:
    std::vector<std::vector<int>> data;
public:
    MultiDimensionalArray(std::vector<std::vector<int>> d) : data(d) {}

    // 重载加法运算符
    MultiDimensionalArray operator+(const MultiDimensionalArray& other) const {
        std::vector<std::vector<int>> result;
        for (int i = 0; i < data.size(); i++) {
            std::vector<int> row;
            for (int j = 0; j < data[i].size(); j++) {
                row.push_back(data[i][j] + other.data[i][j]);
            }
            result.push_back(row);
        }
        return MultiDimensionalArray(result);
    }

    // 重载减法运算符
    MultiDimensionalArray operator-(const MultiDimensionalArray& other) const {
        std::vector<std::vector<int>> result;
        for (int i = 0; i < data.size(); i++) {
            std::vector<int> row;
            for (int j = 0; j < data[i].size(); j++) {
                row.push_back(data[i][j] - other.data[i][j]);
            }
            result.push_back(row);
        }
        return MultiDimensionalArray(result);
    }

    // 重载乘法运算符
    MultiDimensionalArray operator*(const MultiDimensionalArray& other) const {
        std::vector<std::vector<int>> result;
        for (int i = 0; i < data.size(); i++) {
            std::vector<int> row;
            for (int j = 0; j < data[i].size(); j++) {
                row.push_back(data[i][j] * other.data[i][j]);
            }
            result.push_back(row);
        }
        return MultiDimensionalArray(result);
    }

    // 获取多维数组的元素
    int getElement(int row, int col) const {
        return data[row][col];
    }
};

int main() {
    std::vector<std::vector<int>> arr1 = {{1, 2}, {3, 4}};
    std::vector<std::vector<int>> arr2 = {{5, 6}, {7, 8}};

    MultiDimensionalArray a(arr1);
    MultiDimensionalArray b(arr2);

    MultiDimensionalArray result1 = a + b;
    MultiDimensionalArray result2 = a - b;
    MultiDimensionalArray result3 = a * b;

    std::cout << "Addition Result: " << result1.getElement(0, 0) << ", " << result1.getElement(0, 1) << std::endl;
    std::cout << "Subtraction Result: " << result2.getElement(1, 0) << ", " << result2.getElement(1, 1) << std::endl;
    std::cout << "Multiplication Result: " << result3.getElement(1, 0) << ", " << result3.getElement(1, 1) << std::endl;

    return 0;
}