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

最新推荐文章于 2017-04-23 17:42:00 发布

豫帝哥哥

最新推荐文章于 2017-04-23 17:42:00 发布

阅读量1.1k

点赞数

分类专栏： ======【数☆学】====== 欧拉函数费马小定理矩阵快速幂快速幂 HDOJ☚☚☚☚☚☚ ACM-Road

本文链接：https://blog.csdn.net/ydd97/article/details/47683213

版权

HDOJ☚☚☚☚☚☚ 同时被 3 个专栏收录

175 篇文章 0 订阅

订阅专栏

======【数☆学】======

92 篇文章 0 订阅

订阅专栏

ACM-Road

29 篇文章 1 订阅

订阅专栏

M斐波那契数列

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

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

Sample Output

Source

2013金山西山居创意游戏程序挑战赛——初赛（2）

Recommend

liuyiding | We have carefully selected several similar problems for you: 5390 5389 5388 5387 5386

题目分析：

此题是求的相乘的斐波那契数，开始看成求相加斐波那契，以为是很裸的矩阵快速幂。。。

不过，此题虽然让求的是相乘的过程。但是当你写出几个数之后就会发现这些数的指数是成斐波那契数列的。

3 4 12 48..... ==> 3^1*4^0 3^0*4^1 3^1*4^1 3^1*4^2.........

我们可以求出让求的第n项指数是多少。但是因为让求的数比较多，所以结果肯定会比较大，我们要想到对指数取余。我们想到费马小定理：

a ^ n = a ^ (n % euler ( M ) )*( mod M )

题目中说的是结果对M=1000000007取余，所以我们对指数取余的数应为euler(M)。因为M是素数，所以euler(M)=1000000006。

下面我们很容易就能找到相乘的矩阵，进行快速幂。不过注意题目的n，题目n代表的是n+1项。

开始的时候我找的相乘的矩阵是3*3的矩阵。。。。开始TLE了，以为是矩阵的原因，改成了2*2的。。。逗比了。。其实不是矩阵太大。。多了一维而已慢不了哪里。。主要是应该long long型的，写int就TLE。。。long long就过了。

AC代码。。3*3的相乘矩阵

#include <stdio.h>
#include <math.h>
#include <vector>
#include <queue>
#include <string>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>

using namespace std;

const int Modn = 1000000007;
const int MAXN = 3;
typedef long long LL;
LL res[MAXN][MAXN];

void Matmul(LL X[MAXN][MAXN],LL Y[MAXN][MAXN])
{
    int t[MAXN][MAXN]={0};
    for(int i=0;i<MAXN;i++){
        for(int k=0;k<MAXN;k++){
            if(X[i][k])
                for(int j=0;j<MAXN;j++)
                t[i][j]=(t[i][j]+X[i][k]*Y[k][j]%(Modn-1))%(Modn-1);
        }
    }
    for(int i=0;i<MAXN;i++)
        for(int j=0;j<MAXN;j++)
        X[i][j]=t[i][j];
}

void Matrix(LL X[MAXN][MAXN],LL n)
{
    for(int i=0;i<MAXN;i++){
        //res[i][i]=1;
        for(int j=0;j<MAXN;j++)
            res[i][j]=(i==j);
    }
    while(n){
        if(n&1) Matmul(res,X);
        Matmul(X,X);
        n>>=1;
    }
}

LL power(LL x,LL n)
{
    x%=Modn;
    LL res=1;
    while(n){
        if(n&1) res=res*x%Modn;
        x=x*x%Modn;
        n>>=1;
    }
    return res;
}

int main()
{

    LL a,b,n;
    while(scanf("%lld%lld%lld",&a,&b,&n)!=EOF){
        if(n<3){
            if(n==0)printf("%lld\n",a);
            else if(n==1) printf("%lld\n",b);
            else if(n==2) printf("%lld\n",a*b);
            continue;
        }
        LL num[MAXN][MAXN];
        LL f[3];
        memset(num,0,sizeof(num));
        num[0][0]=num[0][1]=1;
        num[1][2]=num[1][0]=1;
        Matrix(num,n-2);//(n+1)-3;
        f[2]=1;f[1]=0;f[0]=1;
        LL ans1=0;
        for(int j=0;j<3;j++)
            ans1=(ans1+f[j]*res[j][0]%(Modn-1))%(Modn-1);
        f[2]=0;f[1]=1;f[0]=1;
        LL ans2=0;
        for(int j=0;j<3;j++)
            ans2=(ans2+f[j]*res[j][0]%(Modn-1))%(Modn-1);
        LL cnt=power(a,ans1)*power(b,ans2)%Modn;
        printf("%lld\n",cnt);
    }
    return 0;
}

2*2的相乘矩阵

#include <stdio.h>
#include <math.h>
#include <vector>
#include <queue>
#include <string>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>

using namespace std;

const int Modn = 1000000007;
const int MAXN = 2;
typedef long long LL;
LL res[MAXN][MAXN];

void Matmul(LL X[MAXN][MAXN],LL Y[MAXN][MAXN])
{
    LL t[MAXN][MAXN]={0};
    for(int i=0;i<MAXN;i++){
        for(int k=0;k<MAXN;k++){
            if(X[i][k])
                for(int j=0;j<MAXN;j++)
                t[i][j]=(t[i][j]+X[i][k]*Y[k][j]%(Modn-1))%(Modn-1);
        }
    }
    for(int i=0;i<MAXN;i++)
        for(int j=0;j<MAXN;j++)
        X[i][j]=t[i][j];
}

void Matrix(LL X[MAXN][MAXN],LL n)
{
    for(int i=0;i<MAXN;i++){
        //res[i][i]=1;
        for(int j=0;j<MAXN;j++)
            res[i][j]=(i==j);
    }
    while(n){
        if(n&1) Matmul(res,X);
        Matmul(X,X);
        n>>=1;
    }
}

LL power(LL x,LL n)
{
    x%=Modn;
    LL ans=1;
    while(n){
        if(n&1) ans=ans*x%Modn;
        x=x*x%Modn;
        n>>=1;
    }
    return ans;
}

int main()
{

    LL a,b,n;
    while(scanf("%lld%lld%lld",&a,&b,&n)!=EOF){
        if(n<3){
            if(n==0)printf("%lld\n",a);
            else if(n==1) printf("%lld\n",b);
            else if(n==2) printf("%lld\n",a*b);
            continue;
        }
        LL num[MAXN][MAXN];
        LL f[3];
        num[0][0]=num[1][0]=num[0][1]=1;
        num[1][1]=0;
        Matrix(num,n-1);//(n+1)-3;
        f[1]=1;f[0]=0;
        LL ans1=0;
        for(int j=0;j<MAXN;j++)
            ans1=(ans1+f[j]*res[j][0]%(Modn-1))%(Modn-1);
        //printf("%d ",ans1);
        f[1]=0;f[0]=1;
        LL ans2=0;
        for(int j=0;j<MAXN;j++)
            ans2=(ans2+f[j]*res[j][0]%(Modn-1))%(Modn-1);
        //printf("%d \n",ans2);
        LL cnt=power(a,ans1)*power(b,ans2)%Modn;
        printf("%lld\n",cnt);
    }
    return 0;
}