蓝桥杯——矩阵乘法

问题描述

给定一个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.util.Scanner;

public class Main {
	public static int[][] twoMatrixPlus(int x [][],int y [][]) {//两个矩阵相乘,具有普适性
		int ans[][]=new int[x.length][y[0].length];
		 for(int i=0;i<x.length;i++){
			 for(int j=0;j<y[0].length;j++){
				 ans[i][j]=0;   //初始化
                 for(int p=0;p<x[0].length;p++)
                  ans[i][j]+=x[i][p]*y[p][j];
             }
         }
		 return ans;          
	}
	public static int[][] matrixPower(int a[][],int n){  //矩阵的幂运算
		int ans[][]=new int[a.length][a[0].length];
		for(int i=0;i<ans.length;i++)
			for(int j=0;j<ans[0].length;j++) {
				if(i==j)
					ans[i][j]=1;
				else
					ans[i][j]=0;
			}//初始化为单位矩阵
		while(n!=0) {
			if((n&1)==1)
				ans=twoMatrixPlus(ans,a);
			a=twoMatrixPlus(a,a);
			n=n>>1;
		}
		return ans;
	}
	
	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();
			}
		}
		b = matrixPower(a,m);
		for(int i = 0; i < n; i++) {
			for(int j = 0; j < n; j++) {
				System.out.print(b[i][j]+" ");
			}
			System.out.println();
		}
	}
}