hdu4549

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

代码:

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<cmath>
#include<algorithm>
typedef long long ll;
const int mod=1000000007;
const int mmod=1000000006;
struct mat{
    int r,c;
    ll m[5][5];
    void clear(){
        for(int i=1;i<=r;i++)memset(m[i],0,sizeof(m[i]));
    }
};

int read(){
    int x=0;
    char c=getchar();
    while(c>'9'||c<'0')c=getchar();
    while(c>='0'&&c<='9'){
        x=x*10+c-'0';
        c=getchar();
    }
    return x;
}

mat MatMul(mat &m1,mat &m2){
    mat tmp;
    tmp.r=m1.r;
    tmp.c=m2.c;
    int i,j,k;
    for(i=1;i<=tmp.r;i++){
        for(j=1;j<=tmp.c;j++){
            ll t=0;
            for(k=1;k<=m1.c;k++){
                t=(t+(m1.m[i][k]*m2.m[k][j])%mmod)%mmod;
            }
            tmp.m[i][j]=t;
        }
    }
    return tmp;
}

mat MatQP(mat &a,int n){
    mat ans,tmp=a;
    ans.r=ans.c=a.r;
    memset(ans.m,0,sizeof(ans.m));
    for(int i=1;i<=ans.r;i++){
        ans.m[i][i]=1;
    }
    while(n){
        if(n&1)ans=MatMul(ans,tmp);
        n>>=1;
        tmp=MatMul(tmp,tmp);
    }
    return ans;
}

ll QP(ll a,ll n){
    ll tmp=a,ans=1;
    while(n){
        if(n&1)ans=ans*tmp%mod;
        tmp=tmp*tmp%mod;
        n>>=1;
    }
    return ans%mod;
}

int main(){
    int x,y,n;
    mat t,tmp;
    t.r=2;t.c=2;
    t.clear();
    t.m[1][2]=t.m[2][1]=t.m[2][2]=1;
    mat a;
    a.r=2;
    a.c=1;
    a.m[1][1]=0;
    a.m[2][1]=1;
    ll ans;
    while(scanf("%d%d%d",&x,&y,&n)!=EOF){
        if(n==0)printf("%d\n",x);
        else{
            tmp=MatQP(t,n-1);  //这里一定要注意指数一定大于等于1的。
            tmp=MatMul(tmp,a);
            ans=QP(x,tmp.m[1][1])*QP(y,tmp.m[2][1])%mod;
            printf("%lld\n",ans);
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_40859951/article/details/88895339