Java基础练习之矩阵乘法

题目：
问题描述
　　给定一个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的值
输出格式
　　输出共N行，每行N个整数，表示A的M次幂所对应的矩阵。相邻的数之间用一个空格隔开
样例输入
2 2
1 2
3 4
样例输出
7 10
15 22

二：代码
两个坑：
第一，矩阵的0次幂是单位矩阵。
第二，每次计算一次乘法时，用一个中间矩阵存储结果，当本次运算结束后，再赋给结果矩阵。

import java.util.Scanner;
public class Main {
	public static void main(String[] args){
		Scanner in = new Scanner(System.in);
		int n = in.nextInt();
		int m = in.nextInt();
		int[][] matrix= new int[n][n];
		for(int i=0;i<n;i++)
		{
			for(int j=0;j<n;j++)
			{
				matrix[i][j]=in.nextInt();
			}
		}
		matrixMultifyM(matrix,n,m);
	}
	public static void matrixMultifyM(int[][]matrix,int n,int m)
	{
		int[][] answer = new int[n][n];
		int[][] temp = new int[n][n];  //作为中间变量
		if(m==0)
		{
			for(int i=0;i<n;i++)
			{
				for(int j=0;j<n;j++)
					answer[i][j]=1;
			}
		}  //0次幂为单位矩阵
		else
		{
			int mm=1;  //乘法次数递增
			while(mm<=m)
			{
				if(mm==1)
				{
					for(int i=0;i<n;i++)
					{
						for(int j=0;j<n;j++)
						{
							answer[i][j]=matrix[i][j];	
						}
					}
				}  //1次幂即本身
				else
				{
					for(int i=0;i<n;i++)
					{
						for(int j=0;j<n;j++)
						{
							int k=n;
							int num=0;
							while(k!=0)
							{
								num+=answer[i][k-1]*matrix[k-1][j];
								k--;
							}
							temp[i][j]=num;  //若此处用answer[][]存储，导致下一次循环上面的answer值不是所期望的运算值
						}
					}
					for(int i=0;i<n;i++)  //存储本次乘法的结果
					{
						for(int j=0;j<n;j++)
						{
							answer[i][j]=temp[i][j];
						}
					}
					
				}
				mm++;
			}
		}
		for(int i=0;i<n;i++)
		{
			for(int j=0;j<n;j++)
			{
				System.out.print(answer[i][j]+" ");
			}
			System.out.println("");
		}
	}
}