在C++中迭代大小可变的N维矩阵

#include <iostream>

// 定义一个N维矩阵类
template <typename T>
class NDimensionalMatrix {
private:
    int* dimensions;  // 存储每个维度的大小
    T* data;          // 存储矩阵数据
    int totalSize;    // 矩阵总大小

public:
    // 构造函数，传入每个维度的大小
    NDimensionalMatrix(int* dims, int numDims) {
        dimensions = new int[numDims];
        totalSize = 1;
        for (int i = 0; i < numDims; i++) {
            dimensions[i] = dims[i];
            totalSize *= dims[i];
        }
        data = new T[totalSize];
    }

    // 析构函数，释放内存
    ~NDimensionalMatrix() {
        delete[] dimensions;
        delete[] data;
    }

    // 获取指定位置的元素
    T& operator()(int* indices) {
        int index = 0;
        int multiplier = 1;
        for (int i = 0; i < numDims; i++) {
            index += indices[i] * multiplier;
            multiplier *= dimensions[i];
        }
        return data[index];
    }
};

int main() {
    int dims[] = {3, 4, 2};  // 定义一个3x4x2的矩阵
    NDimensionalMatrix<int> matrix(dims, 3);

    // 设置矩阵元素的值
    for (int i = 0; i < 3; i++) {
        for (int j = 0; j < 4; j++) {
            for (int k = 0; k < 2; k++) {
                int indices[] = {i, j, k};
                matrix(indices) = i + j + k;
            }
        }
    }

    // 输出矩阵元素的值
    for (int i = 0; i < 3; i++) {
        for (int j = 0; j < 4; j++) {
            for (int k = 0; k < 2; k++) {
                int indices[] = {i, j, k};
                std::cout << "matrix(" << i << ", " << j << ", " << k << ") = " << matrix(indices) << std::endl;
            }
        }
    }

    return 0;
}