POJ Matrix Power Series （矩阵快速幂+二分）

题意：矩阵求和

分析：如果用矩阵快速幂直接把每一项求出来相加会超时。可利用公式，使用二分

快速求出     

（1）当时，

（2）当时，那么有

    

（3）当时，那么有

  

代码：

#include <iostream>
#include <cstdio>

using namespace std;

struct Matrax
{
	int mat[100][100];
}U,F;

int N,MOD,K;

Matrax add(Matrax a,Matrax b)
{
	Matrax ans;
	for(int i=0;i<N;i++)
		for(int j=0;j<N;j++)
			ans.mat[i][j]=(a.mat[i][j]%MOD+b.mat[i][j]%MOD)%MOD;
	return ans;
}

Matrax multi(Matrax a,Matrax b)
{
	Matrax ans;
	for(int i=0;i<N;i++)
	{
		for(int j=0;j<N;j++)
		{
			ans.mat[i][j]=0;
			for(int k=0;k<N;k++)
				ans.mat[i][j]+=(a.mat[i][k]*b.mat[k][j])%MOD;
			ans.mat[i][j]%=MOD;
		}
	}
	return ans;
}

Matrax mkpow(int n)
{
	Matrax ans=U,p=F;
	while(n)
	{
		if(n&1)
			ans=multi(ans,p);
		p=multi(p,p);
		n/=2;
	}
	return ans;
}

Matrax solve(int k)
{
	if(k==1)
		return F;
	Matrax ans=solve(k/2);
	if(k&1)
	{
		Matrax c=multi(ans,mkpow(k/2+1));
		c=add(c,mkpow(k/2+1));
		ans=add(ans,c);
	}
	else
	{
		Matrax c=multi(ans,mkpow(k/2));
		ans=add(ans,c);
	}
	return ans;
}

int main()
{
	int i,j;
	while(cin>>N>>K>>MOD)
	{
		for(i=0;i<N;i++)
			for(j=0;j<N;j++)
			{
				scanf("%d",&F.mat[i][j]);
				F.mat[i][j]%=MOD;
				U.mat[i][j]=(i==j);
			}
		Matrax ans=solve(K);
		for(i=0;i<N;i++)
		{
			for(j=0;j<N;j++)
				printf("%d ",ans.mat[i][j]);
			printf("\n");
		}
	}
	return 0;
}