POJ 3233 Matrix Power Series 矩阵快速幂+二分求和

矩阵快速幂,请参照模板 http://www.cnblogs.com/pach/p/5978475.html

直接sum=A+A2+A3...+Ak这样累加肯定会超时,但是

sum=A+A2+...+Ak/2+A(k/2)*(A+A2+...+Ak/2)    k为偶数时;

sum=A+A2+...+A(k-1)/2+A((k-1)/2)*(A+A2+...+A(k-1)/2)+Ak    k为奇数时。

然后递归二分求和

PS:刚开始mat定义的是__int64,于是贡献了n次TLE。。。

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std; int n,m;
const int N=; struct Mat
{
int mat[N][N];
};
Mat Multiply(Mat a, Mat b)
{
Mat c;
memset(c.mat, , sizeof(c.mat));
for(int k = ; k < n; ++k)
for(int i = ; i < n; ++i)
if(a.mat[i][k])
for(int j = ; j < n; ++j)
if(b.mat[k][j])
c.mat[i][j] = (c.mat[i][j] +a.mat[i][k] * b.mat[k][j])%m;
return c;
}
Mat QuickPower(Mat a, int k)
{
Mat c;
memset(c.mat,,sizeof(c.mat));
for(int i = ; i < n; ++i)
c.mat[i][i]=;
for(; k; k >>= )
{
if(k&) c = Multiply(c,a);
a = Multiply(a,a);
}
return c;
}
Mat Add(Mat a,Mat b)
{
for(int i=; i<n; i++)
for(int j=; j<n; j++)
a.mat[i][j]=(a.mat[i][j]+b.mat[i][j])%m;
return a;
}
Mat Solve(Mat a,int k)
{
if(k==)
return a;
Mat e,ret;
memset(e.mat,,sizeof(e.mat));
for(int i=; i<n; i++)
e.mat[i][i]=;
ret=Multiply(Add(e,QuickPower(a,k>>)),Solve(a,k>>));
if(k%)
return Add(ret,QuickPower(a,k));
return ret;
}
int main()
{
//freopen("in.txt","r",stdin);
int k;
scanf("%d%d%d",&n,&k,&m);
Mat a;
for(int i=; i<n; i++)
for(int j=; j<n; j++)
scanf("%d",&a.mat[i][j]);
Mat ans=Solve(a,k);
for(int i=; i<n; i++)
{
for(int j=; j<n-; j++)
printf("%d ",ans.mat[i][j]);
printf("%d\n",ans.mat[i][n-]);
}
return ;
}
上一篇:POJ 3233 Matrix Power Series(矩阵等比求和)


下一篇:POJ3233 Matrix Power Series 矩阵乘法