POJ 3233 快速矩阵乘法

觉得专业课学习有点落下了，而且acm这方面也没有弄好= =蛋疼死了！！稍微写写博客就开始学习吧！！

矩阵的快速幂和a^b%n类似。运用二分的思想。在这题中需要计算的是A+A^2+A^3+...+A^k的和，由于矩阵相乘有结合律，所以!@#$%^&*.....

不多说了，代码很丑。还是拿来占个位置吧.... 第二次写递归啊！！我的神啊~~

#include<iostream>
#define MAXN 31
using namespace std;

struct Matrix
{
unsigned long long matrix[MAXN][MAXN];
};

int n,m;

void MatrixPow( int k,Matrix &pre )
{
int i,j,p;
if( k<=1 )return ;
else if( k==2 )
{
Matrix now1=pre;
for( i=1;i<=n;i++ )
for( j=1;j<=n;j++ )
{
pre.matrix[i][j]=0;
for( p=1;p<=n;p++ )
pre.matrix[i][j]+=now1.matrix[i][p]*now1.matrix[p][j];
pre.matrix[i][j]%=m;
}
}
else if( k%2==0 )
{
Matrix now1=pre;
MatrixPow( k>>1,now1 );
MatrixPow( 2,now1 );
pre=now1;
}
else
{
Matrix now1=pre;
Matrix now2=pre;
MatrixPow( k>>1,now1 );
MatrixPow( 2,now1 );
for( i=1;i<=n;i++ )
for( j=1;j<=n;j++ )
{
pre.matrix[i][j]=0;
for( p=1;p<=n;p++ )
pre.matrix[i][j]+=now1.matrix[i][p]*now2.matrix[p][j];
pre.matrix[i][j]%=m;
}
}
}

void addMatrix( int k,Matrix &pre )
{
int i,j,p;
if( k<=1 )
return ;
else if( k==2 )
{
Matrix now1=pre;
Matrix now2=pre;
MatrixPow( 2,now2 );
for( i=1;i<=n;i++ )
for( j=1;j<=n;j++ )
{
pre.matrix[i][j]=now1.matrix[i][j]+now2.matrix[i][j];
pre.matrix[i][j]%=m;
}
}
else if( k%2==0 )
{
Matrix now1=pre;
Matrix now2=pre;
addMatrix( k>>1,now1 );
MatrixPow( k>>1,now2 );

for( i=1;i<=n;i++ )
for( j=1;j<=n;j++ )
{
pre.matrix[i][j]=0;
for( p=1;p<=n;p++ )
pre.matrix[i][j]+=now1.matrix[i][p]*now2.matrix[p][j];
pre.matrix[i][j]%=m;
}

for( i=1;i<=n;i++ )
for( j=1;j<=n;j++ )
{
pre.matrix[i][j]+=now1.matrix[i][j];
pre.matrix[i][j]%=m;
}
}
else if( k%2==1 )
{
Matrix now1=pre;
Matrix now2=pre;
addMatrix( k-1,now1 );
MatrixPow( k,now2 );
for( i=1;i<=n;i++ )
for( j=1;j<=n;j++ )
{
pre.matrix[i][j]=now1.matrix[i][j]+now2.matrix[i][j];
pre.matrix[i][j]%=m;
}
}
}

int main()
{
int k;
while( scanf( "%d %d %d",&n,&k,&m )!=EOF )
{
int i,j;
Matrix ans;
for( i=1;i<=n;i++ )
for( j=1;j<=n;j++ )
{
scanf( "%llu",&ans.matrix[i][j] );
}
addMatrix( k,ans );

for( i=1;i<=n;i++ )
{
for( j=1;j<n;j++ )
printf( "%llu ",ans.matrix[i][j]%m );
printf( "%llu\n",ans.matrix[i]
%m );
}
}
return 0;
}