题目分析:构造矩阵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;
}