用Java实现矩阵结构

public class Matrix {
    private int[][] matrix;

    public Matrix(int rows, int cols) {
        matrix = new int[rows][cols];
    }

    public int get(int row, int col) {
        return matrix[row][col];
    }

    public void set(int row, int col, int value) {
        matrix[row][col] = value;
    }

    public int getRowCount() {
        return matrix.length;
    }

    public int getColCount() {
        return matrix[0].length;
    }

    public Matrix add(Matrix other) {
        if (getRowCount() != other.getRowCount() || getColCount() != other.getColCount()) {
            throw new IllegalArgumentException("Matrices must have the same dimensions");
        }

        Matrix result = new Matrix(getRowCount(), getColCount());
        for (int i = 0; i < getRowCount(); i++) {
            for (int j = 0; j < getColCount(); j++) {
                result.set(i, j, get(i, j) + other.get(i, j));
            }
        }
        return result;
    }

    public Matrix multiply(Matrix other) {
        if (getColCount() != other.getRowCount()) {
            throw new IllegalArgumentException("Number of columns in the first matrix must be equal to the number of rows in the second matrix");
        }

        Matrix result = new Matrix(getRowCount(), other.getColCount());
        for (int i = 0; i < getRowCount(); i++) {
            for (int j = 0; j < other.getColCount(); j++) {
                int sum = 0;
                for (int k = 0; k < getColCount(); k++) {
                    sum += get(i, k) * other.get(k, j);
                }
                result.set(i, j, sum);
            }
        }
        return result;
    }
}