hdu 1588 Gauss Fibonacci（矩阵乘法+二分）

题目分析：构造矩阵a={{0,1},{1,1}},{y,x+y}=a*{x,y};

{ans, --}=求和（0->n-1）{(a^k)^i}*a^b*{0,1}.......

正确代码：

以下代码有错误！！！

#include<iostream>
#include<cstdio>
using namespace std;

struct node
{
	int matrix[3][3];
};
int k,b,n,m;
node operator +(node x,node y)
{
     node z;
	 for(int i=1;i<=2;i++)
		 for(int j=1;j<=2;j++)
		 {
			 z.matrix[i][j]=(x.matrix[i][j]+y.matrix[i][j])%m;
		 }
	 return z;
}
node operator * (node x,node y)
{
	node z;
	for(int i=1;i<=2;i++)
		for(int j=1;j<=2;j++)
		{
			int temp=0;
			for(int p=1;p<=2;p++)
			{
				temp+=(x.matrix[i][k]*y.matrix[k][j])%m;
				temp%=m;
			}
			z.matrix[i][j]=temp;
		}
	return z;
}

node pow(node x,int i)//矩阵x^i
{
	node temp;
	memset(temp.matrix,0,sizeof(temp.matrix));
	for(int j=1;i<=2;i++)
		temp.matrix[j][j]=1;
	if(i==0)
		return temp;
	if(i==1)
		return x;

	node y=pow(x,i/2);
	if(i%2==0)
		return y*y;
	else
		return y*y*x;
}
node func(node A,int i)//A^0+A^1+A^2+...A^i
{
    node temp;
	memset(temp.matrix,0,sizeof(temp.matrix));
	for(int j=1;i<=2;i++)
		temp.matrix[j][j]=1;
	if(i==0)
		return temp;
	node B=func(A,(i-1)/2);
	node C=pow(A,(i+2)/2);
	if(i%2==1)
		return (temp+C)*B;
	else
		return temp+(A+C)*B;
}
int main()
{
	while(scanf("%d %d %d %d",&k,&b,&n,&m)!=EOF)
	{
         node a;
		 for(int i=1;i<=2;i++)
			 for(int j=1;j<=2;j++)
				 if(i==1&&j==1)
					 a.matrix[i][j]=0;
				 else
					 a.matrix[i][j]=1;
		 node temp=pow(a,k);
		 node ans=func(temp,n-1);
		 ans=ans*pow(a,b);
		 int res=ans.matrix[1][2];
		 printf("%d\n",res);
	}
	system("pause");
	return 0;
}