poj 3233 矩阵乘法累加

题目就是求一个矩阵的1~k次幂相加

经典题目

代码如下

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<cstdlib>
#include<map>
#include<vector>
#include<algorithm>
using namespace std;
struct mat
{
    int x[100][100];
}A,B,E;
int n,k,mod;
mat operator *(mat a,mat b)
{
    mat c;
    for(int i=0;i<2*n;i++)
    {
        for(int j=0;j<2*n;j++)
        {
            c.x[i][j]=0;
            for(int k=0;k<2*n;k++)
            {
                c.x[i][j]+=(a.x[i][k]*b.x[k][j]);
               c.x[i][j]%=mod;
            }
        }
    }
    return c;
}
mat operator ^(mat x,int y)
{
    mat b=x,r=E;
    while(y)
    {
        if(y&1)
        r=r*b;
        b=b*b;
        y>>=1;
    }
    return r;

}
int main()
{
   // freopen("kepin.txt","r",stdin);
   // freopen("output.txt","w",stdout);

   while(~scanf("%d%d%d",&n,&k,&mod))
   {
       memset(A.x,0,sizeof(A.x));
       memset(B.x,0,sizeof(B.x));
       memset(E.x,0,sizeof(E.x));
       for(int i=0;i<2*n;i++)
       {
           for(int j=0;j<2*n;j++)
           {
               if(i==j)
               E.x[i][j]=1;
           }
       }
       for(int i=0;i<n;i++)
       {
           for(int j=0;j<n;j++)
           {
               scanf("%d",&A.x[i][j]);
           }
       }
       for(int i=0;i<n;i++)
       {
           for(int j=0;j<n;j++)
           {
               B.x[i][j]=A.x[i][j];
           }
           B.x[n+i][i]=B.x[n+i][n+i]=1;
       }
       B=B^(k+1);
       for(int i=0;i<n;i++)
       {
           for(int j=0;j<n;j++)
           {
               int a=B.x[i+n][j]%mod;
               if(i==j)
               a=(a+mod-1)%mod;
               printf("%d%c",a,j==n-1?'\n':' ');
           }
       }
   }
    return 0;
}