矩阵构造 - hdu3306

Another kind of Fibonacci

题意：定义另一种类似斐波那契数列的数列，A(0) = 1 , A(1) = 1 , A(N) = X * A(N - 1) + Y * A(N - 2) (N >= 2)，现在要计算S(N) , S(N) = A(0)^2 +A(1)^2+……+A(n)^2.

数据范围：
每组样例给定 N, X , Y.
N : 2<= N <= 2^31 – 1
X : 2<= X <= 2^31– 1
Y : 2<= Y <= 2^31 – 1

题解：
因为N达到了1e9，所以用矩阵快速幂计算，难点在于构造矩阵。
给出一个大神的题解，以及关于构造矩阵的一些练习。

下为解题报告：林大陈宇

该题为矩阵连乘问题，主要考虑以下公式：

f(n)=x*f(n-1)+y*f(n-2)

先看 s(n)=s(n-1)+f(n)^2;

右边出现了n，不可以的，只能出现n-1  或n-2

所以：s(n-1)=s(n-2)+f(n-1)^2;  右边有2项，s(n-2)和f(n-1)^2必须要有了

  S(n-2)-----------s(n-1)

   F(n-1)^2------f(n)^2=[xf(n-1)+yf(n-2)]^2=出现了f(n-2)^2和2xyf(n-1)f(n-2),所以共4项

最左边是1个空的矩阵(4*4)，也是我们要构造的矩阵（只要学过矩阵的人，就能把它算出来了吧）

 

然后代入矩阵连乘幂的模板算下就行了（该模板要会背着敲）

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <math.h>
#define LL long long

using namespace std;

const LL mod=10007;
struct node
{
    LL mp[4][4];
} res,a;

node mul(node aa,node bb)
{
    int n=4;
    node c;
    memset(c.mp,0,sizeof(c.mp));
    for(int k=0; k<n; ++k)
        for(int i=0; i<n; ++i)
        {
            if(aa.mp[i][k]==0) continue;
            for(int j=0; j<n; ++j)
                c.mp[i][j]=(c.mp[i][j]+(aa.mp[i][k]*bb.mp[k][j])%mod)%mod;
        }
    return c;
}

LL _pow(LL k)
{
    while(k>0)
    {
        if(k&1) res=mul(res,a);
        k/=2;
        a=mul(a,a);
    }

    LL ans=0;
    for(int i=0;i<2;++i)
        for(int j=0;j<4;++j)
        ans=(ans+res.mp[i][j])%mod;

    return ans;
}

int main()
{
    LL n,x,y;
    while(scanf("%lld%lld%lld",&n,&x,&y)!=EOF)
    {
        memset(res.mp,0,sizeof(res.mp));
        res.mp[0][0]=res.mp[1][1]=res.mp[2][2]=res.mp[3][3]=1;

        memset(a.mp,0,sizeof(a.mp));
        a.mp[0][0]=a.mp[0][1]=a.mp[2][1]=1;
        a.mp[1][1]=x*x%mod;
        a.mp[1][2]=y*y%mod;
        a.mp[1][3]=2*x*y%mod;
        a.mp[3][1]=x%mod;
        a.mp[3][3]=y%mod;

        printf("%lld\n",_pow(n-1));

    }
    return 0;
}

其他的扩展练习，在链接：http://pan.baidu.com/s/1bp6L4lT  密码：ao8x  (因为公式太多，写进来真的太麻烦了）