AcWing 3534. 矩阵幂
原题链接
简单
#include<bits/stdc++.h>
using namespace std;
const int N = 10;
int t,n,m,k,l,r,op,x,y;
int a[N][N][N];
void solve(){
cin>>n>>k;
for(int i = 1;i<=n;i++){
for(int j = 1;j<=n;j++){
cin>>a[1][i][j];
}
}
for(int o = 2;o<=k;o++){
for(int i = 1;i<=n;i++){
for(int j = 1;j<=n;j++){
for(int y = 1;y<=n;y++){
a[o][i][j] += a[o-1][i][y] * a[1][y][j];
}
}
}
}
for(int i = 1;i<=n;i++){
for(int j = 1;j<=n;j++){
cout<<a[k][i][j]<<" ";
}
cout<<"\n";
}
// 1 2
}
int main(){
ios::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
solve();
return 0;
}