java矩阵乘法

试题 基础练习 矩阵乘法

资源限制
时间限制：1.0s 内存限制：512.0MB
问题描述
　　给定一个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 矩阵乘法 {
	public static void main(String[] args) {
		Scanner in=new Scanner(System.in);
		int n=in.nextInt(),m=in.nextInt();
		int d[][] = new int[n+1][n+1];
		for(int i=1;i<=n;i++) {
			for(int j=1;j<=n;j++) {
				d[i][j]=in.nextInt();
			}
		}
		int dp1[][] =d;
		if(m==0) {
			for(int i=1;i<=n;i++) {
				for(int j=1;j<=n;j++) {
					if(i==j)d[i][j]=1;
					else d[i][j]=0;
				}
			}
			for(int i=1;i<=n;i++) {
				for(int j=1;j<=n;j++) {
					System.out.print(d[i][j]+" ");
				}
				System.out.println();
			}
		}
		else f(d,dp1,0,n,m);
	}
	private static void f(int[][] d, int[][] dp1, int cur,int n, int m) {
		//d不变的原数组,dp1动态规划,cur判断幂次够不够
		if(cur>=m-1) {
			for(int i=1;i<=n;i++) {
				for(int j=1;j<=n;j++) {
					System.out.print(dp1[i][j]+" ");
				}
				System.out.println();
			}
			return;
		}
		int dp2[][] = new int[n+1][n+1];
		for(int i=1;i<=n;i++) {
			for(int j=1;j<=n;j++) {
				for(int k=1;k<=n;k++) {
					dp2[i][j]+=d[i][k]*dp1[k][j];
				}
			}
		}
		f(d,dp2,cur+1,n,m);
	}
}

感谢关注，点赞一键三连哦！！