题目描述
给定一个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
输出
153 96 108 81
import java.util.*;
import java.io.*;
public class Main{
public static int[][] power(int[][] matrix,int k){//计算矩阵幂的函数
int n = matrix.length;//取得矩阵的维数
int[][] result = new int[n][n];//申请一个用于放结果的数组
for(int i=0;i<n;i++){
System.arraycopy(matrix[i],0,result[i],0,n);//将当前数组的值拷贝到result数组中
}
for(int num=0;num<k-1;num++){//循环k次幂
for(int i=0;i<n;i++){//行循环
int[] row = new int[n];//将行循环的结果暂时放在数组中
for(int j=0;j<n;j++){//列循环
int sum =0;
for(int m=0;m<n;m++){//行列相乘,用m进行循环
sum += result[i][m] * matrix[m][j];
}
row[j] = sum;//将结果放到暂时存放的数组中,此时result中的行已经计算完,不再使用
}
System.arraycopy(row,0,result[i],0,n);//将暂存数组中的元素拷贝到结果数组中,整行拷贝
}
}
return result;
}
public static void printMat(int[][] matrix){//打印结果函数
int n = matrix.length;
for(int i=0;i<n;i++){//行循环
StringBuilder sb = new StringBuilder();//申请一个stringbuilder,存放将要输出的结果
for(int j=0;j<n;j++){//列循环
if(j != n-1){
sb.append(matrix[i][j]).append(" ");//不是最后一个元素值的,后面添加空格
}else{
sb.append(matrix[i][j]);//最后一个元素,后面不加空格
}
}
System.out.println(sb);//按行打印
}
}
public static void main(String[] args) {
Scanner sn = new Scanner(System.in);
while(sn.hasNextInt()){
int n = sn.nextInt();//输入向量维数
int k = sn.nextInt();//输入幂
int[][] mat = new int[n][n];
for(int j=0;j<n;j++){
for(int m=0;m<n;m++){
mat[j][m] = sn.nextInt();//输入矩阵
}
}
int[][] result = power(mat,k);
printMat(result);
}
}
}
(参考别人的代码)