hdoj 4549 M斐波那契数列 【矩阵快速幂 + 费马小定理 + 快速幂】

M斐波那契数列 Time Limit: 3000/1000 MS (Java/Others)    Memory Limit: 65535/32768 K (Java/Others) Total Submission(s): 2176    Accepted Submission(s): 629 Problem Description M斐波那契数列F[n]是一种整数数列，它的定义如下： F[0] = a F[1] = b F[n] = F[n-1] * F[n-2] ( n > 1 ) 现在给出a, b, n，你能求出F[n]的值吗？   Input 输入包含多组测试数据； 每组数据占一行，包含3个整数a, b, n（ 0 <= a, b, n <= 10^9 ）   Output 对每组测试数据请输出一个整数F[n]，由于F[n]可能很大，你只需输出F[n]对1000000007取模后的值即可，每组数据输出一行。   Sample Input 0 1 0 6 10 2   Sample Output 0 60

思路比较好想吧，没什么弯。但犯二的我对矩阵取余取错了，更狗血的是测试数据可以过。。。害我WA两次！

分析：用MOD表示1000000007

首先根据题意我们可以得到F[0] = a 、 F[1] = b 、F[2] = a*b 、 F[3] = a * b^2 、 F[4} = a^2 * b^3、F[5] = a^3 * b^5...

再来看fibonacci数列f[0] = 1、f[1] = 1、f[3] = 2、f[4] = 3 、f[5] = 5、f[6] = 8...

结论——F[i]项a的系数是f[i-2]，b的系数是f[i-1]。

这样就是求解a ^ (f[i-2]) * b ^ (f[i-1]) % MOD，拆开后a^f[i-2] % MOD * b^f[i-1] % MOD。

由于指数过大，不能直接快速幂，我们可以采用费马小定理优化一下。

费马小定理推导公式a^n = a ^ (n % (m-1)) (%m)。成立条件a是整数，m是质数且a不是m的倍数。

回到a^f[i-2]%MOD * b^f[i-1] % MOD，由于MOD > a、b，且MOD是质数，满足公式成立条件。

思路：

一、先用矩阵快速幂求解f[n-1]和f[n-2]，切记取余是对MOD的欧拉函数即MOD-1操作！！！；

二、求出f[i-1]和f[i-2]后，用费马小定理优化指数得到a的系数X和b的系数Y；

三、对a^X * b^Y直接快速幂操作。

AC代码：

#include <cstdio>
#include <cstring>
#define LL long long
#define MOD 1000000007
struct Matrix
{
    LL a[3][3];
};
Matrix ori, res;
LL F[3];
void init()
{
    memset(ori.a, 0, sizeof(ori.a));
    memset(res.a, 0, sizeof(res.a));
    for(int i = 0; i < 2; i++)
        res.a[i][i] = 1;
    ori.a[0][1] = ori.a[1][0] = ori.a[1][1] = 1;
}
Matrix muitl(Matrix x, Matrix y)
{
    Matrix z;
    memset(z.a, 0, sizeof(z.a));
    for(int i = 0; i < 2; i++)
    {
        for(int k = 0; k < 2; k++)
        {
            if(x.a[i][k] == 0) continue;
            for(int j = 0; j < 2; j++)//醉了 取余错了， WA两次
                z.a[i][j] = (z.a[i][j] + (x.a[i][k] * y.a[k][j]) % (MOD-1)) % (MOD-1);
        }
    }
    return z;
}
LL pow_mod(LL a, LL n)
{
    LL ans = 1;
    while(n)
    {
        if(n & 1)
            ans = ans * a % MOD;
        a = a * a % MOD;
        n >>= 1;
    }
    return ans;
}
LL ans[3];//a和b的系数
void solve(LL a, LL b, LL n)
{
    init();
    while(n)
    {
        if(n & 1)
            res = muitl(ori, res);
        ori = muitl(ori, ori);
        n >>= 1;
    }
    for(int i = 0; i < 2; i++)
    {
        ans[i] = 0;
        for(int k = 0; k < 2; k++)
            ans[i] = (ans[i] + (res.a[i][k] * F[k]) % (MOD-1)) % (MOD-1);
    }
    LL X = ans[0] % (MOD-1);//a的指数 费马小定理优化
    LL Y = ans[1] % (MOD-1);//b的指数
    LL A = pow_mod(a, X);//快速幂
    LL B = pow_mod(b, Y);
    printf("%lld\n", A*B%MOD);
}
int main()
{
    LL a, b, n;
    F[0] = F[1] = 1;//构建初始fibonacci
    while(scanf("%lld%lld%lld", &a, &b, &n) != EOF)
    {
        if(n == 0)
        {
            printf("%lld\n", a % MOD);
            continue;
        }
        if(n == 1)
        {
            printf("%lld\n", b % MOD);
            continue;
        }
        solve(a, b, n-2);//第n项a、b系数对应fibonacci的n-1项和n-2项
    }
    return 0;
}