M斐波那契数列 - HDU 4549 费马小定理

M斐波那契数列

Time Limit: 3000/1000 MS (Java/Others)    Memory Limit: 65535/32768 K (Java/Others)
Total Submission(s): 1990    Accepted Submission(s): 570

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
  
  

 

思路：设f[]为斐波那契数，F[n]为a^f[n-1]*b^f[n]，由费马小定理a^(p-1)%p == 1，我们将斐波那契数用矩阵快速幂取模1e9+6，得到f[n-1]和f[n]，然后用快速幂得出最后结果。

AC代码如下：

#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
struct node
{
    ll f[2][2];
};
ll mod=1e9+7;
int n;
node mul(node a,node b)
{
    int i,j,k;
    node m;
    for(i=0;i<=1;i++)
       for(j=0;j<=1;j++)
       {
           m.f[i][j]=0;
           for(k=0;k<=1;k++)
              m.f[i][j]=(m.f[i][j]+a.f[i][k]*b.f[k][j])%(mod-1);
       }
    return m;
}
ll POW(ll num,ll k)
{
    int i,j;
    ll first=num,ret=1;
    while(k)
    {
        if(k&1)
          ret=ret*first%mod;
        first=first*first%mod;
        k/=2;
    }
    return ret;
}
int main()
{
    int i,j,k;
    ll a,b,ans;
    node first,ret;
    while(~scanf("%I64d%I64d%d",&a,&b,&n))
    {
        if(n==0)
          printf("%I64d\n",a%mod);
        else if(n==1)
          printf("%I64d\n",b%mod);
        else
        {
            k=n-2;
            first.f[0][0]=1;first.f[0][1]=1;
            first.f[1][0]=1;first.f[1][1]=0;

            ret.f[0][0]=1;ret.f[0][1]=1;
            ret.f[1][0]=0;ret.f[1][1]=0;

            while(k)
            {
                if(k&1)
                  ret=mul(ret,first);
                first=mul(first,first);
                k/=2;
            }
            ans=POW(a,ret.f[0][1])*POW(b,ret.f[0][0])%mod;
            printf("%I64d\n",ans);
        }
    }
}