自定义C++ Array类以支持多维数组

下面是一个简单的C++ Array类，用于支持多维数组：

#include <vector>

template <typename T>
class Array {
private:
    std::vector<T> data;
    std::vector<size_t> shape;

public:
    Array(const std::vector<size_t>& shape) : shape(shape) {
        size_t size = 1;
        for (size_t s : shape) {
            size *= s;
        }
        data.resize(size);
    }

    T& operator()(const std::vector<size_t>& indices) {
        size_t index = 0;
        size_t multiplier = 1;
        for (size_t i = 0; i < indices.size(); i++) {
            index += indices[i] * multiplier;
            multiplier *= shape[i];
        }
        return data[index];
    }
};

使用示例：

int main() {
    Array<int> arr({2, 3});
    arr({0, 0}) = 1;
    arr({0, 1}) = 2;
    arr({0, 2}) = 3;
    arr({1, 0}) = 4;
    arr({1, 1}) = 5;
    arr({1, 2}) = 6;

    std::cout << arr({0, 0}) << std::endl;
    std::cout << arr({0, 1}) << std::endl;
    std::cout << arr({0, 2}) << std::endl;
    std::cout << arr({1, 0}) << std::endl;
    std::cout << arr({1, 1}) << std::endl;
    std::cout << arr({1, 2}) << std::endl;

    return 0;
}

这个Array类实现了一个简单的多维数组，可以通过传入不同维度的shape来创建不同维度的数组。可以通过传入一个包含各个维度下标的向量来访问和修改数组中的元素。