Another kind of Fibonacci(矩阵)

Another kind of Fibonacci

Time Limit: 3000/1000 MS (Java/Others)    Memory Limit: 65536/65536 K (Java/Others)
Total Submission(s): 1919    Accepted Submission(s): 738

Problem Description

As we all known , the Fibonacci series : F(0) = 1, F(1) = 1, F(N) = F(N - 1) + F(N - 2) (N >= 2).Now we define another kind of Fibonacci : A(0) = 1 , A(1) = 1 , A(N) = X * A(N - 1) + Y * A(N - 2) (N >= 2).And we want to Calculate S(N) , S(N) = A(0)² +A(1)²+……+A(n)².

 

Input

There are several test cases.
Each test case will contain three integers , N, X , Y .
N : 2<= N <= 2³¹ – 1
X : 2<= X <= 2³¹– 1
Y : 2<= Y <= 2³¹ – 1

 

Output

For each test case , output the answer of S(n).If the answer is too big , divide it by 10007 and give me the reminder.

 

Sample Input

2 1 1

3 2 3

 

Sample Output

6

196

 

题意：很明确了。。。不多说。

思路：

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3306

转载请注明出处：寻找&星空の孩子

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
#define LL __int64
#define mod 10007

struct matrix
{
    LL mat[][];
};

matrix multiply(matrix a,matrix b)
{
    matrix c;
    memset(c.mat,,sizeof(c.mat));
    for(int i=;i<;i++)
    {
        for(int j=;j<;j++)
        {
            if(a.mat[i][j]==)continue;
            for(int k=;k<;k++)
            {
                if(b.mat[j][k]==)continue;
                c.mat[i][k]=(c.mat[i][k]+a.mat[i][j]*b.mat[j][k])%mod;
            }
        }
    }
    return c;
}

matrix quickmod(matrix a,LL n)
{
    matrix res;
    for(int i=;i<;i++)
        for(int j=;j<;j++)
            res.mat[i][j]=(i==j);
    while(n)
    {
        if(n&) res=multiply(res,a);
        n>>=;
        a=multiply(a,a);
    }
    return res;
}

int main()
{
    LL n,x,y;

    while(scanf("%I64d%I64d%I64d",&n,&x,&y)!=EOF)
    {
        if(n==||n==)
        {
            printf("%I64d\n",n+);
            continue;
        }
        matrix ans;
        ans.mat[][]=ans.mat[][]=;
        ans.mat[][]=ans.mat[][]=ans.mat[][]=;
        ans.mat[][]=ans.mat[][]=ans.mat[][]=;
        ans.mat[][]=ans.mat[][]=x*x%mod;
        ans.mat[][]=ans.mat[][]=y*y%mod;
        ans.mat[][]=ans.mat[][]=*x*y%mod;
        ans.mat[][]=x%mod;
        ans.mat[][]=y%mod;

        ans=quickmod(ans,n-);
        printf("%I64d\n",(*ans.mat[][]+ans.mat[][]+ans.mat[][]+ans.mat[][])%mod);

    }
    return ;
}

加油！少年！