A1041. 矩阵乘法

问题描述

　　给定一个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

package www.tsinsen.com;

import java.util.Scanner;

public class A1041 {

    public static void main(String[] args) {
        // TODO Auto-generated method stub
        Scanner scan=new Scanner(System.in);
        int n=scan.nextInt();
        int m=scan.nextInt();
        int[][] arr=new int[n][n];//存放原始数组
        int[][] temp=new int[n][n];//存放n-1次幂的结果
        int[][] result=new int[n][n];
        //输入
        for(int i=0;i<n;i++) {
            for(int j=0;j<n;j++) {
                arr[i][j]=scan.nextInt();
                temp[i][j]=arr[i][j];
                result[i][j]=arr[i][j];
            }
        }
        //如果是0次幂输出单位矩阵
        for(int i=0;i<n;i++) {
            for(int j=0;j<n;j++) {
                if(i==j) {
                    result[i][j]=1;
                }
                else{
                    result[i][j]=0;
                }
            }
        }
        //幂运算
        for(int k=1;k<m;k++) {
            for(int i=0;i<n;i++) {
                for(int j=0;j<n;j++) {
                    //矩阵运算
                    int sum=0;
                    for(int l=0;l<n;l++) {
                        sum+=temp[i][l]*arr[l][j];
                    }
                    result[i][j]=sum;
                }
            }
            //把result的结果赋给temp
            for(int i=0;i<n;i++) {
                for(int j=0;j<n;j++) {
                    temp[i][j]=result[i][j];
                }
            }
        }
        //输出
        for(int i=0;i<n;i++) {
            for(int j=0;j<n;j++) {
                //判读是不是每行中的最后一个元素，不是的话加空格，使得话不加
                if(j!=n-1) {
                    System.out.print(result[i][j]+" ");
                }else {
                    System.out.println(result[i][j]);
                }
            }
        }
    }

}