描述
给定一个n*n的矩阵,求该矩阵的k次幂,即P^k。
输入描述:
第一行:两个整数n(2<=n<=10)、k(1<=k<=5),两个数字之间用一个空格隔开,含义如上所示。 接下来有n行,每行n个正整数,其中,第i行第j个整数表示矩阵中第i行第j列的矩阵元素Pij且(0<=Pij<=10)。另外,数据保证最后结果不会超过10^8。
输出描述:
对于每组测试数据,输出其结果。格式为: n行n列个整数,每行数之间用空格隔开,注意,每行最后一个数后面不应该有多余的空格。
示例1
输入:
2 2 9 8 9 3 3 3 4 8 4 9 3 0 3 5 7 5 2 4 0 3 0 1 0 0 5 8 5 8 9 8 5 3 9 6 1 7 8 7 2 5 7 3
输出:
153 96 108 81 1216 1248 708 1089 927 504 1161 1151 739 47 29 41 22 16 147 103 73 116 94 162 108 153 168 126 163 67 112 158 122 152 93 93 111 97
代码如下:
import java.util.Scanner;/** 矩阵幂*/
public class MatrixPower {public static void main(String[] args) {Scanner scanner = new Scanner(System.in);while(scanner.hasNext()) {int n = scanner.nextInt();int k = scanner.nextInt();int[][] arr = new int[n][n];for (int i = 0; i < n; i++) {for (int j = 0; j < n; j++) {arr[i][j] = scanner.nextInt();}}int[][] a = arr;for (int t = 0; t < k-1; t++) {//temp矩阵为arr矩阵的t+2次方int[][] temp = new int[n][n];//矩阵乘法的得到的结果保存到temp中for (int i = 0; i < n; i++) {for (int j = 0; j < n; j++) {for (int k1 = 0; k1 < n; k1++) {temp[i][j] += a[i][k1] * arr[k1][j];}}}a = temp;}for(int i = 0;i < n; i++) {for(int j = 0;j < n; j++) {System.out.print(a[i][j]+" ");}System.out.println();}}}}