fib(限制内存)矩阵+快速幂

Fib数列是这样的数列: 1,1,2,3,5,8,13,21,,,,,,. 你能确定它的第n项吗,假设它的第n项为N,你需要编程解决对于另一个输入数M,N % M的值.

思路:矩阵临时存储

代码:

#include
#include
#include
using namespace std;
int n;
long long mod;
struct Matrix
{
    long long m[2][2];
}M;
Matrix Mult(Matrix a,Matrix b)
{
    Matrix ans;
    for(int i=0;i     {
        for(int j=0;j         {
            ans.m[i][j]=0;
            for(int k=0;k             {
                ans.m[i][j]+=a.m[i][k]*b.m[k][j]%mod;
                ans.m[i][j]%=mod;
            }
        }
    }
    return ans;
}
Matrix quickpow(Matrix a,long long b)
{
    Matrix ans;
    for(int i=0;i     {
        for(int j=0;j         {
            if(i==j)
                ans.m[i][j]=1;
            else
                ans.m[i][j]=0;
        }
    }
    while(b)
    {
        if(b&1)
            ans=Mult(ans,a);
        a=Mult(a,a);
        b/=2;
    }
    return ans;
}
int main()
{
    int t;
    scanf("%d%lld",&t,&mod);
    while(t--)
    {
        Matrix N,M;
        memset(N.m,0,sizeof(N.m));
        long long m;
        scanf("%lld",&m);
 
        M.m[0][0]=1;M.m[0][1]=1;
        M.m[1][0]=1;M.m[1][1]=0;
        N.m[0][0]=2;
        N.m[1][0]=1;
 
        if(m==1||m==2)
        {
            long long ans=1;
            printf("%lld\n",ans%mod);
            continue;
        }
        n=2;
        M=quickpow(M,m-3);
        N=Mult(M,N);
        printf("%lld\n",N.m[0][0]);
    }
return 0;
}




























































































































你可能感兴趣的:(快速幂取模)