基础练习矩阵乘法？

问题描述：给定一个N阶矩阵A，输出A的M次幂（M是非负整数）
　　例如：
　　A =
　　1 2
　　3 4
　　A的2次幂
　　7 10
　　15 22
输入描述:
　　第一行是一个正整数N、M（1<=N<=30, 0<=M<=5），表示矩阵A的阶数和要求的幂数
　　接下来N行，每行N个绝对值不超过10的非负整数，描述矩阵A的值
输入样例:
2 2
1 2
3 4
输出描述:
　　输出共N行，每行N个整数，表示A的M次幂所对应的矩阵。相邻的数之间用一个空格隔开
输出样例:
7 10
15 22
代码记录：
import java.util.Scanner;
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];
    for(int i=0;i<n;i++) {
        for(int j=0;j<n;j++) {
            a[i][j]=sc.nextInt();
        }
    }
    for(int k=1;k<m;k++) {

//控制m次方

        int[][] c=new int[n][n];

//暂存数组，用于累加

        for(int i=0;i<n;i++) {
            for(int j=0;j<n;j++) {
                for(int w=0;w<n;w++) {

//控制所乘位置下标

                    if(k==1) {

//计算二次方时，用于存放结果的数组b没有初值，可以直接累加

                        b[i][j]+=a[w][j]*a[i][w];
                    }
                    else {

//在不改变前几次乘方结果和所乘原始矩阵的情况下，进行相乘累加

                        c[i][j]+=b[w][j]*a[i][w];
                    }
                }
            }
        }
        if(k>1) {

//在完整地进行每一次矩阵相乘后，把最终结果赋给数组b

            for(int i=0;i<n;i++) {
                for(int j=0;j<n;j++) {
                    b[i][j]=c[i][j];
                }
            }
        }
    }
    if(m==1) {

//矩阵1次方

        for(int i=0;i<n;i++) {
            for(int j=0;j<n;j++) {
                b[i][j]=a[i][j];
            }
        }
    }
    if(m==0) {

//矩阵0次方

        for(int i=0;i<n;i++) {
            b[i][i]=1;
        }
    }
    for(int i=0;i<n;i++) {
        for(int j=0;j<n;j++) {
            System.out.print(b[i][j]+"\t");
        }
        System.out.println();
    }
}

}
结果记录：