题目描述:
给定n*n的矩阵A,求A^k
输入格式:
第一行,n,k
第2至n+1行,每行n个数,第i+1行第j个数表示矩阵第i行第j列的元素
输出格式:
输出A^k
共n行,每行n个数,第i行第j个数表示矩阵第i行第j列的元素,每个元素模10^9+7
样例输入:
2 1 1 1 1 1
样例输出:
1 1
1 1
代码如下:
#include<bits/stdc++.h>
using namespace std;
long long n,a[101][101]={};
long long s[101][101]={};
int const x=1e9+7;
void ju(long long b[101][101],long long c[101][101]){
long long d[101][101]={};
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
for(int ii=1;ii<=n;ii++){
d[i][j]+=(b[i][ii]*c[ii][j])%x;
d[i][j]=d[i][j]%x;
}
}
}
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
b[i][j]=d[i][j];
}
}
}
void kuai(long long k){
for(int i=1;i<=n;i++){
s[i][i]=1;
}
while(k){
if(k&1){
ju(s,a);
}
ju(a,a);
k/=2;
}
}
int main(){
long long k;
cin>>n>>k;
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
cin>>a[i][j];
}
}
kuai(k);
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
cout<<s[i][j]<<" ";
}cout<<endl;
}
return 0;
}