矩阵乘法——蓝桥练习

题目：
给定一个N阶矩阵A，输出A的M次幂（M是非负整数）
例如：
　　A =
　　1 2
　　3 4
　　
　A的2次幂
　 7 10
　15 22
　输入格式　　
第一行是一个正整数N、M（1<=N<=30, 0<=M<=5），表示矩阵A的阶数和要求的幂数接下来N行，每行N个绝对值不超过10的非负整数，描述矩阵A的值
　输出格式　　
输出共N行，每行N个整数，表示A的M次幂所对应的矩阵。相邻的数之间用一个空格隔开

#include <iostream>
#include <cmath>
using namespace std;
int main()
{
 int i,j,k,n,m,a[31][31],ra[31][31],ta[31][31];
 cin>>n>>m;
 for(i=1;i<=n;i++)
 {
  for(j=1;j<=n;j++)
  {
   cin>>a[i][j];
   ta[i][j]=a[i][j];
  }
 }
 if(m==1)
 {
  cout<<pow(a[1][1],m);
 }
 else if(m==0)
 {
  for(i=1;i<=n;i++)
  {
   for(j=1;j<=n;j++)
   {
    if(i==j)
    {
     cout<<1<<" ";
    }
    else {
     cout<<0<<" ";
    }
   }
   cout<<endl;
  }
 }
 else{
  m=m-1;
  while(m--)
  {
   for(i=1;i<=n;i++)
   {
    for(j=1;j<=n;j++)
    {
     ra[i][j]=0;
     for(k=1;k<=n;k++)
     {
      ra[i][j]+=ta[i][k]*a[k][j];
     }
    }
   }
   for(i=1;i<=n;i++)
   {
    for(j=1;j<=n;j++)
    {
     ta[i][j]=ra[i][j];
    }
   }
  }
  for(i=1;i<=n;i++)
  {
   for(j=1;j<=n;j++)
   {
    cout<<ta[i][j]<<" ";
   }
   cout<<endl;
  }
 }
 return 0;
}