[caioj 1488及hdu 1588]Gauss Fibonacci

有一个g数组，f数组。g[i]=k*i+b，k与i都是给定的常数。f[0]=0,f[1]=1,f[i]=f[i-1]+f[i-2] (i>=2)。现在，要你求f[g[0]]+f[g[1]]+…+f[g[n-1]]的值，但可能最后的值会很大，所以结果需要mod M。

这题看到了它的数据范围，就令人很容易想到用矩阵乘法加快速幂，但是这是一道有难度的题，耗了我不少时间。它的难点就是如何构造矩阵，对于这个问题，网上有两种做法，而本蒟蒻给出一种做法，是巧妙地运用到了矩阵乘法的结合律的，如果用这种方法就像是pku 3233和pku 3070的结合，大家可以先做一下这两题，我的blog也详解了这两题。首先我们求出构造出Fibonacci数列的矩阵a，为{0,1}{1,1}。为什么，因为a * {f1}{f2}=(f2}{f3(f1+f2)}（两个相邻括号之间有空行），然后初始序列s为{0}{1}， 因为a^x * {0}{1}={f[x]}{f[x+1]}。
这样我们就可以用两次快速幂来分别算出A矩阵（表示a^b),B矩阵（表示a^k)，而f[g[0]]+f[g[1]]+…+f[g[n-1]]可以看为f[b]+f[k+b]+…+f[k * (n-1)+b] ,则上式可为[A+a^(k+b)+…+a^(k^(n-1)+b)] * s，提公因式，则变成[a^b * (1+a^k+…+(a^k)^(n-1))] * s，转换后的公式为[A*(1+B+B^2+…+B^(n-1))] * s。对于B+B^2+…+B^(n-1)可以用两次快速幂做，详见http://blog.csdn.net/lixuanjing2016/article/details/77235674
搞定B+B^2+…+B^(n-1)后，就可以算出转换后的公式，这题就迎刃而解了。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
struct node
{
    long long a[5][5];
    node()
    {
        memset(a,0,sizeof(a));
    }
};
int M;
node jiafa(node a,node b)
{
    node c;
    for(int i=1;i<=2;i++)
    {
        for(int j=1;j<=2;j++)
        {
            c.a[i][j]=(c.a[i][j]+a.a[i][j]+b.a[i][j])%M;
        }
    }
    return c;
}
node chengfa1(node a,node b)
{
    node c;
    for(int i=1;i<=2;i++)
    {
        for(int j=1;j<=2;j++)
        {
            for(int k=1;k<=2;k++)
            {
                c.a[i][j]=(c.a[i][j]+a.a[i][k]*b.a[k][j])%M;
            }
        }
    }
    return c;
}
node chengfa2(node a,node b)
{
    node c;
    for(int i=1;i<=2;i++)
    {
        for(int k=1;k<=2;k++)
        {
            c.a[i][1]=(c.a[i][1]+a.a[i][k]*b.a[k][1])%M;
        }
    }
    return c;
}
node power(node a,int x)
{
    node ans;
    ans.a[1][1]=ans.a[2][2]=1;
    while(x>0)
    {
        if(x%2==1)ans=chengfa1(a,ans);
        a=chengfa1(a,a);
        x/=2;
    }
    return ans;
}
node getsum(node a,int k)
{
    if(k==1)return a;
    node t;
    if(k%2==0)
    {
        t=getsum(a,k/2);
        t=jiafa(t,chengfa1(t,power(a,k/2)));
    }
    else
    {
        t=getsum(a,k/2);
        t=jiafa(t,chengfa1(t,power(a,k/2)));
        t=jiafa(t,power(a,k));
    }
    return t;
}
int main()
{
    int k,b,n,x;
    while(scanf("%d%d%d%d",&k,&b,&n,&M)!=EOF)
    {
        n--;
        node f,ans,ss,per1,per2,s;
        f.a[1][1]=0;f.a[2][1]=1;
        ss.a[1][1]=ss.a[2][2]=1;
        s.a[1][2]=s.a[2][1]=s.a[2][2]=1;
        per1=power(s,k);
        per2=power(s,b);
        ans=chengfa1(per2,jiafa(getsum(per1,n),ss));
        //单位矩阵ss就相当于1的存在
        f=chengfa2(ans,f);
        printf("%lld\n",f.a[1][1]);
    }
    return 0;
}