矩阵幂

题目描述

给定一个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);
            
        }
    }
}

（参考别人的代码）