基础练习 矩阵乘法

问题描述

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

import java.text.DecimalFormat;
import java.text.NumberFormat;
import java.util.ArrayList;
import java.util.Scanner;  

    public class Main{   
        public static void main(String[] args){  
            Scanner input = new Scanner(System.in);
            int n,m;
            n = input.nextInt();
            m = input.nextInt();
            int[][] a = new int[n][n];
            int[][] temp = 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] = input.nextInt();
                    b[i][j] = a[i][j];
                }
            }
            int h = m;
            int sum = 0;
            int i,j;
            if(h==0){
                for(i=0;i<n;i++){
                    for(j=0;j<n;j++){
                        if(i==j)
                            System.out.print("1 ");
                        else
                            System.out.print("0 ");
                    }
                    System.out.println();
                }
                System.exit(0);
            }
            while(--h!=0){
                for(i=0;i<n;i++){
                    for(j=0;j<n;j++){
                        sum = 0;
                        for(int k=0;k<n;k++){
                            sum = sum+a[i][k]*b[k][j];
                        }
                        temp[i][j] = sum;
                    }
                }
                for(i=0;i<n;i++){
                    for(j=0;j<n;j++){
                        a[i][j] = temp[i][j];
                    }
                }
            }
            for(i=0;i<n;i++){
                for(j=0;j<n;j++){
                        System.out.print(a[i][j]+" ");
                }
                System.out.println();
            }
            
        }
 }