矩阵乘法

import java.util.Arrays;
import java.util.Scanner;
import java.util.prefs.BackingStoreException;

public class Main {

	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		
		int n = sc.nextInt();
		int m = sc.nextInt();
		
		int[][] a = new int[n][n];
		int[][] b = new int[n][n];
		int[][] c = new int[n][n];//用来记录结果
		for(int i = 0 ; i < n ; i++  ){
			for(int j = 0 ; j < n; j++ ){
				a[i][j] = sc.nextInt();
				b[i][j] = a[i][j];
			}
		}
		if(m == 0 ){
			for(int i = 0 ; i < n ; i++  ){
				for(int j = 0 ; j < n ; j++ ){
					if(i == j){
						a[i][j] = 1;
					}else{
						a[i][j] = 0 ;
					}
				}
				
			}
		}else{
			int flag = 1;
			for(int i = 0 ; i < m-1 ; i++ ){//此处n-1（因为，乘一次得到的是矩阵a的平方）
				if(flag == 1){
					matrixMultiplication(a, b, c, n);
					flag++;
				}else{
					copyMatrix(c, a, n);//这里写copy函数的原因是：如果a = c （把c的引用给a），那么对c置零时，a中元素也会变为零
					matrixMultiplication(a, b, c, n);
				}
				
			}
			copyMatrix(c, a, n);
		}
		

		
		
		
		for(int i = 0 ; i < n ; i++  ){
			for(int j = 0 ; j < n; j++ ){
				System.out.print(a[i][j]+" ");
			}
			System.out.println();
		}
		
	}
	
	public static void matrixMultiplication(int[][] a, int[][] b, int[][] c,int n){//把矩阵a和矩阵b的乘积放入c矩阵中
		for(int i = 0 ; i < n ; i++  ){
			for(int j = 0 ; j < n; j++ ){
				c[i][j] = 0;
			}
		}
		for(int i = 0 ; i < n ; i++  ){
			for(int j = 0 ; j < n; j++ ){
				for(int k = 0 ; k < n ; k++ ){
					c[i][j] += a[i][k]*b[k][j];
				}
			}
		}
	}
	
	public static  void copyMatrix(int[][] c , int[][] a,int n){//把矩阵c copy 到a矩阵中
		for(int i = 0 ; i < n ; i++ ){
			for(int j = 0 ; j < n ; j++ ){
				a[i][j] = c[i][j];
			}
		}
		
	}
	
}