试题 基础练习 矩阵乘法 Java

蓝桥杯VIP试题 基础练习 矩阵乘法：

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

思路：

根据M次幂 让矩阵与自身初始矩阵乘M-1次。
这就要求 多个二维数组，第一个用来保存最一开始输入的矩阵（自身初始矩阵），第二个用来存储上次相乘得到的结果矩阵，第三个用来保留本次相乘得到的新矩阵
*例如 2x2矩阵的3次幂
1 2
3 4
第一个二维数组保存1234，第二个数组保存1 2 3 4X1 2 3 4的结果即7 10 15 22，第三个数组保存1 2 3 4X7 10 15 22的结果 *

import java.util.Scanner;
public class MatrixMultiplication {
	/*
	 * 给定一个N阶矩阵A，输出A的M次幂（M是非负整数） 例如： A = 1 2 3 4 A的2次幂 7 10 15 22
	 */
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n, m;	//	n阶，m次
		n = sc.nextInt();
		m = sc.nextInt();
		int arr[][] = new int[n][n];//上文提到的第一个数组
		int arr1[][] = new int[n][n];//上文提到的第三个数组  也就是最后输出的数组
		int arr2[][] = new int[n][n];//上文提到的第二个数组
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < n; j++) {
				arr[i][j] = sc.nextInt();
				arr2[i][j] = arr[i][j]; //初始让第一第三一样
			}
		}
		for (int l = 0; l < m - 1; l++) {//相乘次数
			for (int i = 0; i < n; i++) {
				for (int j = 0; j < n; j++) {
					for (int k = 0; k < n; k++) {
						arr1[i][j] += arr2[i][k] * arr[k][j];
						//这里arr1储存本次计算的结果
					}

				}

			}
			for (int i = 0; i < n; i++) {
				for (int j = 0; j < n; j++) {
					arr2[i][j] = arr1[i][j];//arr2接受arr1的结果 并成为下次计算的乘数之一
					if (l < m - 2) {
						arr1[i][j] = 0;//每一次相乘前都让此数组归零
					}
				}
			}
		}
		if (m == 0) {//对于M==0, 0次幂情况，正对角线为1，其余为零
			for (int i = 0; i < n; i++) {
				for (int j = 0; j < n; j++) {
					if (i == j) {
						arr[i][j] = 1;
					} else {
						arr[i][j] = 0;
					}
				}
			}
		}
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < n; j++) {
				if (m < 2) {//0、1次幂
					System.out.print(arr[i][j] + " ");
				} else {

					System.out.print(arr1[i][j] + " ");
				}
			}
			System.out.println();
		}

	}
}