ans数组为最后需要输出的答案。temp数组记录每一次乘积之后的内容。数组a为输入的矩阵,需要乘m次a。
#include <bits/stdc++.h>
using namespace std;
int ans[33][33],temp[33][33],a[33][33];
int main()
{
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
cin>>a[i][j];
for(int i=1;i<=n;i++)//行
for(int j=1;j<=n;j++)//列
ans[i][j]=a[i][j];
if(m)
{
m--;
while(m)//m次幂
{
m--;
memset(temp,0,sizeof(temp));
for(int i=1;i<=n;i++)//temp的行
for(int j=1;j<=n;j++)//temp的列
for(int k=1;k<=n;k++)//
temp[i][j]+=ans[i][k]*a[k][j];
for(int i=1;i<=n;i++)//temp的行
for(int j=1;j<=n;j++)//temp的列
ans[i][j]=temp[i][j];
}
}
else
{
memset(ans,0,sizeof(ans));
for(int i=1;i<=n;i++)
ans[i][i]=1;
}
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
cout<<ans[i][j];
if(j==n)cout<<endl;
else cout<<" ";
}
return 0;
}