表示对于数学渣一样的我这道题还是挺有难度的,此题重点在于如何将f(g(i))转换成能处理的东西:参考
构造一个Fibonacci矩阵A=[ 1 1;1 0] (其实就是Fibonacci的递推矩阵)
A^n的右上角那个元素便是f(n)
令f(n)=A^n
当 i=0,1...n的时候,有:
f(0)=A^b
f(1)=A^(k+b)
f(2)=A^(2*k+b)
...
f(n-1)=A^((n-1)*k+b)
则我们需要求得的和sum为:f(0)+f(1)+...+f(n-1) 这个矩阵的右上角元素
有:ans=f(0)+f(1)+...+f(n-1) = A^b + A^b*A^k + A^b*(A^k)^2+...+A^b*(A^k)^(n-1) = A^b*(E+A^k+(A^k)^2+...+(A^k)^(n-1)) //看起好眼花啊。。。
令B=A^k 也就是ans=A^b*(E+B+B^2+B^3+ . . . + B^(N-1)) //这样好点了,关于括号里的等比矩阵求和可见这里
最后ans的右上角的元素便是答案
#include<stdio.h>
#include<string.h>
#define MAXN 5
#define ll long long
struct matrix
{
ll m[MAXN][MAXN];
};
ll mod;
matrix multi(const matrix &a,const matrix &b,int n)
{
matrix ans;
memset(&ans,0,sizeof(ans));
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
for(int k=0;k<n;k++)
{
ans.m[i][j]+=a.m[i][k]*b.m[k][j];
ans.m[i][j]%=mod;
}
return ans;
}
matrix pow(matrix a,int k,int n)
{
matrix ans;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
if(i==j) ans.m[i][j]=1;
else ans.m[i][j]=0;
while(k)
{
if(k&1) ans=multi(ans,a,n);
a=multi(a,a,n);
k>>=1;
}
return ans;
}
void debug(const matrix &a,int n)
{
printf("\n");
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
printf("%lld ",a.m[i][j]);
printf("\n");
}
printf("\n");
}
int main()
{
matrix a,ans;
int k,b,n;
while(scanf("%d%d%d%lld",&k,&b,&n,&mod)!=EOF)
{
memset(&a,0,sizeof(a));
a.m[0][0]=a.m[0][1]=a.m[1][0]=1;
//debug(a,2);//1
a=pow(a,k,2);
//debug(a,2);//2
a.m[0][2]=a.m[1][3]=a.m[2][2]=a.m[3][3]=1;
//debug(a,4);//3
ans=pow(a,n,4);
//debug(ans,4);//4
for(int i=0;i<2;i++)
for(int j=0;j<2;j++)
ans.m[i][j]=ans.m[i][j+2];
//debug(ans,4);//5
memset(&a,0,sizeof(a));
a.m[0][0]=a.m[0][1]=a.m[1][0]=1;
//debug(a,2);//6
ans=multi(ans,pow(a,b,2),2);
//debug(ans,4);//7
printf("%lld\n",ans.m[0][1]%mod);
}
return 0;
}