蓝桥杯试题矩阵乘法java实现

import java.util.Scanner;

public class Main {

    public static void main(String[] args) {
        Scanner sc=new Scanner(System.in);
        int a=sc.nextInt();
        int n=sc.nextInt();
        int[][] m=new int[a][a];
        int[][] f=new int[a][a];//为了保留 结果多次幂时会用到
        //存入到二维数组中去
        for(int i=0;i<a;i++) {
            for(int k=0;k<a;k++) {
                f[i][k]=m[i][k]=sc.nextInt();
            }
        }
        //判断幂的次数当等于零时对角线上的数字 为一
        if(n==0) {
            for(int i=0;i<a;i++) {
                for(int k=0;k<a;k++) {
                    if(i==k) {
                        System.out.print(1+" ");
                    }else {
                        System.out.print(0+" ");
                    }
                }
                System.out.println();
            }
            //1次幂时原样输出
        }else if(n==1) {
            for(int i=0;i<a;i++) {
                for(int k=0;k<a;k++) {
                    System.out.print(m[i][k]+" ");
                    }
                System.out.println();
                }
        }else {
            //几次幂几次循环
        for(int i=1;i<n;i++) {
            //保留运算结果不会改变 开始 建立的两个数组
            int[][] tmp = new int[a][a];
            for(int     k=0;k<a;k++) {
                for(int l=0;l<a;l++) {
                    int e=0;//用于累加储存结果
                    for(int t=0;t<a;t++) {
                        e+=m[k][t]*f[t][l];
                    }
                    tmp[k][l]=e;//赋值到tmp数组中
                }
            }
            f=tmp;//循环完毕更改保留结果以便于下次幂运算时进行累乘
        }
    //打印结果
        for(int i=0;i<a;i++) {
            for(int k=0;k<a;k++) {
                System.out.print(f[i][k]+" ");
            }
            System.out.println();
        }
        }
    }

}