Java实现矩阵加减乘除转制等运算


Java初学,编写矩阵预算程序,当做工具,以便以后写算法时使用。




public class MatrixOperation{
    public static int[][] add(int[][] matrix_a, int[][] matrix_b){
        int row = matrix_a.length;
        int col = matrix_a[0].length;
        int[][] result = new int[row][col];
        
        if(row != matrix_b.length || col != matrix_b[0].length){
            System.out.println("Fault");
        }
        else{
            for(int i=0;i<row;i++){
                for(int j=0;j<col;j++){
                    result[i][j] = matrix_a[i][j] + matrix_b[i][j];
                }
            }
        }
        return result;
    }


    public static int[][] sub(int[][] matrix_a, int[][] matrix_b){
        int row = matrix_a.length;
        int col = matrix_a[0].length;
        int[][] result = new int[row][col];
        
        if(row != matrix_b.length || col != matrix_b[0].length){
            System.out.println("Fault");
        }
        else{
            for(int i=0;i<row;i++){
                for(int j=0;j<col;j++){
                    result[i][j] = matrix_a[i][j] - matrix_b[i][j];
                }
            }
        }
        return result;
    }


    public static int[][] dot(int[][] matrix_a, int[][] matrix_b){
        /*
            matrix_a's dimention m*p matrix_b's dimention p*n.
            return dimention m*n
        */
        int row = matrix_a.length;
        int col = matrix_a[0].length;
        int[][] result = new int[row][col];
        
        if(col != matrix_b.length){
            System.out.println("Fault");
        }
        else{
            for(int i=0;i<row;i++){
                for(int j=0;j<col;j++){
                    result[i][j] = 0;
                    for (int k=0;k<col;k++){
                        result[i][j] += matrix_a[i][k] * matrix_b[k][j];
                    }
                }
            }
        }
        return result;
    }


    public static int[][] dot(int[][] matrix_a, int b){
        int row = matrix_a.length;
        int col = matrix_a[0].length;
        int[][] result = new int[row][col];
        
        for(int i=0;i<row;i++){
            for(int j=0;j<col;j++){
                result[i][j] = matrix_a[i][j] * b;
            }
        }
        return result;
    }
    
    
    public static int[][] mul(int[][] matrix_a, int[][] matrix_b){
        /*
            matrix_a's dimention m*n matrix_b's dimention m*n.
            return dimention m*n
        */
        int row = matrix_a.length;
        int col = matrix_a[0].length;
        int[][] result = new int[row][col];
        
        if(row != matrix_b.length || col != matrix_b[0].length){
            System.out.println("Fault");
        }
        else{
            for(int i=0;i<row;i++){
                for(int j=0;j<col;j++){
                    result[i][j] = matrix_a[i][j] * matrix_b[i][j];
                }
            }
        }
        return result;
    }
    
    
    public static int[][] transport(int [][] matrix_a){
        int row = matrix_a.length;
        int col = matrix_a[0].length;
        int[][] result = new int[row][col];
        
        for(int i=0;i<row;i++){
            for(int j=0;j<col;j++){
                result[j][i] = matrix_a[i][j];
            }
        }
        return result;
    }


    public static void print(int[][] matrix){
        int row = matrix.length;
        int col = matrix[0].length;
                
        for(int i=0;i<row;i++){
            System.out.print("[");
            for(int j=0;j<col;j++){
                System.out.print(matrix[i][j]);
                if (j != col - 1){
                    System.out.print(", ");
                }
            }
            System.out.print("]\n");
        }
    }


    public static void main(String[] args) {
        int[][] a = {{1,2},{3,4}};
        int[][] b = {{7,8},{6,5}};

        int[][] c = add(a, b);
        System.out.println("matrix a = ");
        print(a);
        System.out.println("matrix b = ");
        print(b);
        System.out.println("matrix a + b = ");
        print(c);

        c = sub(a,b);
        System.out.println("matrix a - b = ");
        print(c);
        
        int[][] d = dot(a, b);
        System.out.println("matrix a dot b = ");
        print(d);

        int[][] e = dot(a, 3);
        System.out.println("matrix a * 3 = ");
        print(e);
        
        int[][] f = transport(a);
        System.out.println("matrix a.T = ");
        print(f);
        
        int[][] g = mul(a, b);
        System.out.println("matrix a * b = ");
        print(g);
    }
}



  • 1
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值