hdu 6050 Funny Function （矩阵快速幂）

二维的一个矩阵。

如果只有一维，用一下矩阵快速幂求前缀和是很简单的。

比如我构造出一个矩阵从[f(1,1）,f(1,2), sum(1,2)]推到[f(1,2), f(1,3), sum(1,3)],这是不难的。

由于我们是要求f(m,1)，就是要求sum(m-1,n)所以还要构造矩阵从f(m-1, 1)推到f(m, 1)也就是sum(m-2，n)到sum(m-1, n)。

但是第一维构造的矩阵没有办法和第二维结合，所以是不行的。

f[i]=f[i-1]+2*f[i-2]是一个常系数线性递推数列，那么它的前缀和也一定是一个常系数线性递推数列，比如斐波那契数列sum(1,n)=f[n+2]-1.

所以我们可以通过**（dabiao)等方式找到一个f[i]的前缀和式子，然后用数学归纳法去证明。（太弱了，没法直接推出来)。

这里的公式就是

所以当n为偶数时，f(i+1，1)可以直接由f(i, j+1)-f(i, 1)推出.

f(i, j+1)可以直接由f(i, 1), f(i, 2)构造矩阵得出:[f1, f2]*A=[f2, f3]，进行n次乘法即可。

所以[f(m,1), f(m, 2)]=[f(m-1, j+1)-f(m-1, 1), f(m-1, j+1)+f(m-1, 1)-f(m-1, 2)-f(m-1, 1)]=[f(m-1, j+1)-f(m-1, 1), f(m-1, j+1)-f(m-1, 2)]

=[f(m-1, j+1), f(m-1, j+2)]-[f(m-1, 1), f(m-1, 2)]

=[f(m-1, 1), f(m-1, 2)]*A*A...-[f(m-1, 1), f(m-1, 2)]

=[f(m-1, 1), f(m-1, 2)]*[A*A*A..+E]

只要对[A*A...+E]用矩阵快速幂做m-1次乘法就可以从f(1, 1)推出f(m, 1)了。

奇数同理。

代码：

#include <bits/stdc++.h>
#define LL long long
using namespace std;
const LL mod=1e9+7;
struct node
{
    LL rec[3][3];
    void init()
    {
        for(int i=0; i<2; i++)
        {
            for(int j=0; j<2; j++)rec[i][j]=0;
        }
    }



};

node operator *(const node &a,const node &b)
{
    node res;
    res.init();
    int i, j, k;
    for(i=0; i<2; i++)
    {
        for(j=0; j<2; j++)
        {
            for(k=0; k<2; k++)
            {
                res.rec[i][j]=(res.rec[i][j]+(a.rec[i][k]*b.rec[k][j])%mod)%mod;
                res.rec[i][j]=((res.rec[i][j]%mod)+mod)%mod;
            }
        }
    }
    return res;
}
node operator +(const node &a, const node &b)
{
    node res;
    res.init();
    int i, j;
    for(i=0; i<2; i++)
    {
        for(j=0; j<2; j++)
        {
            res.rec[i][j]=((a.rec[i][j]+b.rec[i][j])%mod+mod)%mod;
        }
    }
    return res;
}

node rec_mod(node a, LL n)
{
    node res, base=a;
    int i, j;
    res.init();
    for(i=0; i<2; i++)res.rec[i][i]=1;
    while(n)
    {
        if(n&1)res=res*base;
        n>>=1;
        base=base*base;
    }
    return res;
}
int main()
{
    int t, i, j;
    LL n, m;
    node a, b;
    a.rec[0][0]=0, a.rec[0][1]=2;
    a.rec[1][0]=1, a.rec[1][1]=1;

    cin>>t;
    while(t--)
    {
        scanf("%lld%lld", &n, &m);
        if(n&1)
        {
            b.rec[0][0]=1, b.rec[0][1]=-2;
            b.rec[1][0]=-1, b.rec[1][1]=0;
        }
        else
        {
            b.rec[0][0]=-1, b.rec[0][1]=0;
            b.rec[1][0]=0, b.rec[1][1]=-1;
        }

        node base=rec_mod(rec_mod(a, n)+b, m-1);

        node ans;
        ans.init();
        ans.rec[0][0]=1, ans.rec[0][1]=1;
        ans=ans*base;
        printf("%lld\n", ans.rec[0][0]%mod);

    }
    return 0;
}