hdu 1588 Gauss Fibonacci(函数嵌套、转换、等比矩阵求和)

hdu 1588 Gauss Fibonacci

表示对于数学渣一样的我这道题还是挺有难度的，此题重点在于如何将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;
}