一维数组和三维数组使用一个模板

#include <iostream>
#include <vector>

template <typename T, int Dim>
class MultiDimArray {
public:
    MultiDimArray(std::initializer_list<size_t> sizes) {
        size_t total = 1;
        for (size_t size : sizes) {
            total *= size;
            dimensions.push_back(size);
        }
        data.resize(total);
    }

    T& operator()(std::initializer_list<size_t> indices) {
        size_t index = 0;
        size_t multiplier = 1;
        for (int i = indices.size(); i > 0; --i) {
            index += indices.begin()[i - 1] * multiplier;
            multiplier *= dimensions[indices.size() - i];
        }
        return data[index];
    }

private:
    std::vector<T> data;
    std::vector<size_t> dimensions;
};

int main() {
    MultiDimArray<int, 1> arr1d = {5};
    arr1d({2}) = 42;
    std::cout << "1D Array[2] = " << arr1d({2}) << std::endl;

    MultiDimArray<int, 3> arr3d = {2, 3, 4};
    arr3d({1, 2, 3}) = 99;
    std::cout << "3D Array[1][2][3] = " << arr3d({1, 2, 3}) << std::endl;

    return 0;
}