将int数组转换为矩阵类以完成运算符重载

#include <vector>
#include <iostream>

class Matrix {
private:
    std::vector<std::vector<int>> data;
    int rows, cols;

public:
    // 构造函数，接受一个整数数组和矩阵的维度
    Matrix(const std::vector<int>& arr, int rows, int cols) : rows(rows), cols(cols) {
        data.resize(rows, std::vector<int>(cols));
        for (int i = 0; i < rows * cols; ++i) {
            data[i / cols][i % cols] = arr[i];
        }
    }

    // 获取矩阵的行数
    int getRows() const { return rows; }

    // 获取矩阵的列数
    int getCols() const { return cols; }

    // 重载加法运算符
    Matrix operator+(const Matrix& other) const {
        if (rows != other.rows || cols != other.cols) {
            throw std::invalid_argument("Matrix dimensions must match for addition.");
        }
        Matrix result(data, rows, cols);
        for (int i = 0; i < rows; ++i) {
            for (int j = 0; j < cols; ++j) {
                result.data[i][j] += other.data[i][j];
            }
        }
        return result;
    }

    // 打印矩阵
    void print() const {
        for (const auto& row : data) {
            for (int val : row) {
                std::cout << val << " ";
            }
            std::cout << std::endl;
        }
    }
};

int main() {
    std::vector<int> arr = {1, 2, 3, 4, 5, 6};
    Matrix m1(arr, 2, 3);
    Matrix m2(arr, 2, 3);

    Matrix sum = m1 + m2;
    sum.print();

    return 0;
}