HDU 4549 M斐波那契数列(矩阵快速幂&费马小定理)

ps:今天和战友聊到矩阵快速幂,想到前几天学长推荐去刷矩阵专题,挑了其中唯一一道中文题,没想到越过山却被河挡住去路。。。

题目链接:[kuangbin带你飞]专题十九 矩阵 R - M斐波那契数列

Time Limit:1000MS     Memory Limit:32768KB     64bit IO Format:%I64d & %I64u

题意

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取模后的值即可,每组数据输出一行。

思路

一看到题,就拿二维01矩阵变来变去尝试半天,无果,就手推了几部,猛然发现f[i] = a^fib[i- 1]*b^fib[i]。道友们动动手,很容易就能发现。
然后就是刷刷的矩阵快速幂求fib,求出两数,再快速幂,结果得出,俩样例试过没问题,提交,wrong。。。
找来找去,找不出错误在哪,去百度了下,发现别人在对矩阵快速幂的时候都是对MOD-1取余的,顿时纳闷,才发现自己忽略了a^b时迭代时不能直接取余的。
看解释说是由费马小定理可以得出循环节是MOD-1,所以对MOD-1取余。仔细的看了费马小定理,似懂非懂的,至于循环节,压根就没想通,可怜在数论上几近一张白纸的我。(ps:过两天好好研究研究)

代码

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <vector>
#include <cmath>
#include <map>

using namespace std;

#define LL long long

const int MOD = 1000000007;

struct Node
{
    LL a, b, c, d;
    Node(){};
    Node(LL a, LL b, LL c, LL d)
    :a(a),b(b),c(c),d(d){}
    Node operator*(const Node& t)
    {
        Node ans;
        ans.a = ((this->a*t.a)%(MOD-1)+(this->b*t.c)%(MOD-1))%(MOD-1);
        ans.b = ((this->a*t.b)%(MOD-1)+(this->b*t.d)%(MOD-1))%(MOD-1);
        ans.c = ((this->c*t.a)%(MOD-1)+(this->d*t.c)%(MOD-1))%(MOD-1);
        ans.d = ((this->c*t.b)%(MOD-1)+(this->d*t.d)%(MOD-1))%(MOD-1);
        return ans;
    }
};

LL qpower(LL n, LL x)
{
    LL ans = 1;
    LL p = x;
    while(n)
    {
        if(n & 1)
            ans = (ans*p)%MOD;
        p = (p*p)%MOD;
        n >>= 1;
    }
    return ans;
}

int main()
{
    LL a, b, n;
    while(cin>>a>>b>>n)
    {
        Node p = Node(0, 1, 1, 1);
        Node ans = Node(1, 0, 0, 1);
        if(n == 0)
        {
            cout<<a<<endl;
            continue;
        }
        --n;
        while(n)
        {
            if(n & 1)
                ans = ans*p;
            p = p*p;
            n >>= 1;
        }
        LL fiba = ans.b;
        LL fibb = ans.d;
        LL rel = 0;
        rel = (qpower(fiba, a) * qpower(fibb, b))%MOD;
        cout<<rel<<endl;
    }
}

你可能感兴趣的:(HDU 4549 M斐波那契数列(矩阵快速幂&费马小定理))