java矩阵相乘

Java实现矩阵相乘

package com.company;

public class Matrix {
    public int row;
    public int col;
    public double[][] data;
    Matrix(double[][] array){
        //获得矩阵的行与列数
        this.row=array.length;
        this.col=array[0].length;
        //为data申请空间
        this.data=new double[row][col];
        //遍历数组进行数组复制
        for(int i=0;i<row;i++){
            System.arraycopy(array[i], 0, this.data[i], 0, col);
        }
    }
    //矩阵乘法
    public Matrix multiply(Matrix B){
        //构造返回结果矩阵
        Matrix C=new Matrix(new double[this.row][B.col]);
        //遍历A矩阵的行
        for(int A_row=0;A_row<this.row;A_row++){
            //遍历B的列与行相乘求和
            for(int B_col=0;B_col<B.col;B_col++) {
                double sum = 0d;
                for (int A_col = 0; A_col < this.col; A_col++) {
                    sum += this.data[A_row][A_col] * B.data[A_col][B_col];
                }
                C.data[A_row][B_col] = sum;
            }
        }
        return C;
    }
    //重写toString方法
    @Override
    public String toString(){
        String  result="[\n";
        for(int i=0;i<this.row;i++){
            result+="[";
            for(int j=0;j<this.col;j++){
                //将this.data[i][j]保留三位小数,转为字符串+" "
                result+=String.format("%.3f ", this.data[i][j]);
            }
            result+="]\n";
        }
        result+="]";
        return result;
    }
}