BASIC17-矩阵乘法（基础练习）

BASIC17-矩阵乘法

时间限制：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

解题思路

根据矩阵乘法公式

 

 得出A的第1行各个数与B的第1列各个数对应相乘后加起来,就是乘法结果中第1行第1列的数。

若幂数为0，则从左上角到右上角的对角线上的元素均为1其余全为0。

代码部分

public static void main(String[] args) {
		// TODO Auto-generated method stub
		
		Scanner input =new Scanner(System.in);
		int n=input.nextInt();//阶数
		int m=input.nextInt();//幂数
		
		long[][] arr=new long[n][n];//定义一个二维数组
		
		//创建矩阵并赋值
		for (int i = 0; i < arr.length; i++) {
			for (int j = 0; j < arr[0].length; j++) {
				arr[i][j]=input.nextLong();
			}
		}
		
		//当幂数等于0,对角为1，其余为0
		if (m==0) {
			arr=new long[n][n];//初始化数组
			for (int i = 0; i < arr.length; i++) {
				arr[i][i]=1;
			}
		}
		
		
		long[][] temp=arr;//数组
		//幂数
		for (int i = 1; i < m; i++) {
			arr=pow(arr,temp);
		}
		//输出
		for (int i = 0; i < arr.length; i++) {
			for (int j = 0; j < arr[0].length; j++) {
				System.out.print(arr[i][j]+" ");
			}
			System.out.println();
		}
	}

	private static long[][] pow(long[][] arr, long[][] temp) {
		
		
		
		long[][] result=new long[arr.length][arr[0].length];
		
		for (int i = 0; i < result.length; i++) {
			for (int j = 0; j < result.length; j++) {
				//随时在变化的下标
				for (int d = 0; d < result.length; d++) {
					//第i行*第j列
					result[i][j]+=arr[i][d]*temp[d][j];

					}
			}
		}
		
		return result;
	}

  运行结果 

     3 0
    1 2 3
    4 5 6
    7 8 9
    1 0 0 
    0 1 0 
    0 0 1 

---

    2 2
    1 2
    3 4
    7 10 
   15 22 

 

 

2019级王新萍  留