自定义C++ Array类以支持复杂索引

#include <iostream>
#include <vector>

class Array {
private:
    std::vector<int> data;
    int num_rows;
    int num_cols;

public:
    Array(int rows, int cols) : num_rows(rows), num_cols(cols) {
        data.resize(rows * cols);
    }

    int& operator()(int row, int col) {
        return data[row * num_cols + col];
    }

    int getValue(int row, int col) const {
        return data[row * num_cols + col];
    }

    void print() const {
        for (int i = 0; i < num_rows; i++) {
            for (int j = 0; j < num_cols; j++) {
                std::cout << getValue(i, j) << " ";
            }
            std::cout << std::endl;
        }
    }
};

int main() {
    Array arr(3, 3);

    // Initialize array
    for (int i = 0; i < 3; i++) {
        for (int j = 0; j < 3; j++) {
            arr(i, j) = i * 3 + j;
        }
    }

    // Print array
    arr.print();

    return 0;
}