蓝桥网试题 java 基础练习 矩阵乘法

------------------------------------------------------------

第一次感觉到好好学习的重要性QAQ

在做这道题之前请先学会 ：矩阵乘法(百度百科)

矩阵的0次幂：对角线为1 其他值为0 例如

结果

------------------------------------------------------------

 算法

import java.util.*;
public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();
        long[][] a = new long[n][n];
        long[][] b = new long[n][n];
        int i,j;
        for(i=0;i<n;i++)
            for(j=0;j<n;j++)
                b[i][j]=a[i][j] = sc.nextLong();
        if(m==0)
            for(i=0;i<n;i++){
                for(j=0;j<n;j++){
                    if(i==j)System.out.print(1+" ");
                    else System.out.print(0+" ");
                }
                System.out.println();
            }
        else if(m==1)
            for(i=0;i<n;i++){
                for(j=0;j<n;j++)
                    System.out.print(a[i][j]+" ");
                System.out.println();
            }
        else{
            for(int z=1;z<m;z++){
                long[][] tmp = new long[n][n]; 
                for(i=0;i<n;i++){
                    for(j=0;j<n;j++){
                        long add = 0;
                        for(int y=0;y<n;y++){
                            add += a[i][y]*b[y][j];
                        }
                        tmp[i][j] = add;
                    }
                }
                b = tmp;
            }
            for(i=0;i<n;i++){
                for(j=0;j<n;j++)
                    System.out.print(b[i][j]+" ");
                System.out.println();
            }
        }
            
    }
}