矩阵k次幂 采用三重循环

Posted 2020-08-15

#include<iostream>
using namespace std;
        
int main()
{
    int n,k;
    long long a[11][11],b[11][11],c[11][11];
    while(cin>>n>>k)
    {
        int x=k;
 
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                {
                    cin>>a[i][j];
                    b[i][j]=a[i][j];
                    c[i][j]=0;
                }     
    k=k-1;
      while(k--)
      {
 
         for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                for(int m=0;m<n;m++)
                  
                 {
             c[i][j]=a[i][m]*b[m][j]+c[i][j];
             }
               if(k>=1)
       {   
             
            for(int i=0;i<n;i++) 
            for(int j=0;j<n;j++)
            {
                a[i][j]=c[i][j];
                c[i][j]=0;
            }
 
        }
 
       }     
       if(x==1)
       {
         for(int i=0;i<n;i++)
                 {
                    for(int j=0;j<n-1;j++)
                cout<<b[i][j]<<‘ ‘;   
             cout<<b[i][n-1]<<endl;
         }
       }
       else
       {
                  for(int i=0;i<n;i++)
                 {
                    for(int j=0;j<n-1;j++)
             
                cout<<c[i][j]<<‘ ‘;   
             cout<<c[i][n-1]<<endl;
         }
          
     }
     cout<<endl;
    }
    return 0;
}

 c[i][j]=a[i][m]*b[m][j]+c[i][j];
这一句是关键，掌握了矩阵相乘的计算公式，则可得到