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次幂所对应的矩阵。相邻的数之间用一个空格隔开

import java.util.Scanner;


public class 矩阵的幂次方 {

	/**
	 * @param args
	 */
	public static void main(String[] args) {
		// TODO Auto-generated method stub
        Scanner sc = new Scanner(System.in);
        int num = sc.nextInt(); //num*num阶的矩阵
        int cifang = sc.nextInt();  //多少次方
        int arr[][]=new int[num][num];  //定义一个二维数组
        int arrfu[][]=new int[num][num];  //定义一个副本，用于后面的运算
        for (int i = 0; i < num; i++) {
			for (int j = 0; j <num; j++) {
				arr[i][j]=sc.nextInt();
			}
		}
        for (int i = 0; i < num; i++) {
			for (int j = 0; j <num; j++) {
				arrfu[i][j]=arr[i][j];
			}
		}
        int arr1[][]=new int[num][num]; //存放结果
        for (int i = 1; i <cifang; i++) { //循环次数，相当于（循环次数+1）次方
			for (int j = 0; j <num; j++) {
				for (int j2 = 0; j2 <num; j2++) {
					int sum = 0; //用来储存每次相乘后的值
					for (int k = 0; k <num; k++) {
						sum+=arr[j][k]*arrfu[k][j2]; //第“j”行与第"j2"列相乘
					}
					arr1[j][j2]=sum; //结果存入新开辟的数组中
				}
			}
			for (int s = 0; s < num; s++) {  //把循环一次得到的结果重新赋值与原数组
				for (int j = 0; j <num; j++) {
					arr[s][j]=arr1[s][j];
				}
			}
		}
        if (cifang==0) {   //需要考虑0次方
			for (int i = 0; i <num; i++) {
				for (int j = 0; j <num; j++) {
					if (i==j) {  //矩阵的0次方代表着主对角线上的都为1，其他的都为0
						System.out.print(1+" ");
					}else {
						System.out.print(0+" ");
					}
				}
				System.out.println();
			}
		}
        else {
		  for (int i = 0; i <num; i++) {  //输出矩阵
			for (int j = 0; j <num; j++) {
				System.out.print(arr[i][j]+" ");
			}
			System.out.println();
		  }
        }
	}

}

2 2
1 2
3 4

7 10
15 22