POJ 3070 矩阵快速幂解决fib问题

Posted 2020-06-24 ATM

矩阵快速幂：http://www.cnblogs.com/atmacmer/p/5184736.html

 

题目链接

#include<iostream>
#include<cstdio>
using namespace std;
typedef long long ll;
#define MOD 10000

ll a[7],b[7],a0[7],b0[7];

void pow_mod(ll n)
{
    a0[1]=a0[2]=a0[3]=1,a0[4]=0;
    b0[1]=b0[4]=1,b0[2]=b0[3]=0;
    while(n)
    {
        if(n%2==1)
        {
            b[1]=(a0[1]*b0[1]+a0[2]*b0[3])%MOD;
            b[2]=(a0[1]*b0[2]+a0[2]*b0[4])%MOD;
            b[3]=(a0[3]*b0[1]+a0[4]*b0[3])%MOD;
            b[4]=(a0[3]*b0[2]+a0[4]*b0[4])%MOD;
            for(int i=1;i<=4;i++)
                b0[i]=b[i];
        }
        a[1]=(a0[1]*a0[1]+a0[2]*a0[3])%MOD;
        a[2]=(a0[1]*a0[2]+a0[2]*a0[4])%MOD;
        a[3]=(a0[3]*a0[1]+a0[4]*a0[3])%MOD;
        a[4]=(a0[3]*a0[2]+a0[4]*a0[4])%MOD;
        for(int i=1;i<=4;i++)
                a0[i]=a[i];
        n/=2;
    }
}

int main()
{
    ll n;
    while(~scanf("%I64d",&n)&&n!=-1)
    {
        if(n==0) printf("0\\n");
        else
        {
            pow_mod(n);
            printf("%I64d\\n",b[2]);
        }
    }
    return 0;
}