[POJ3070]Fibonacci（矩阵快速幂）

题目描述

传送门

题意
求Fibonacci第n项模10000的值

题解

题面中就给了一种十分优越的矩阵做法。
之前的做法又多加了一个矩阵，改姿势了。

代码

#include
#include
#include
using namespace std;
const int Mod=10000;

int n;
struct hp{int a[3][3];}st,m,ans;

inline hp cheng(hp a,hp b)
{
    hp ans;
    memset(ans.a,0,sizeof(ans.a));
    for (int i=1;i<=2;++i)
        for (int j=1;j<=2;++j)
            for (int k=1;k<=2;++k)
                ans.a[i][j]=(ans.a[i][j]+a.a[i][k]*b.a[k][j]%Mod)%Mod;
    return ans;
}
inline hp matrix_fast_pow(hp a,int p)
{
    hp ans=st;
    for (;p;p>>=1,a=cheng(a,a))
        if (p&1)
            ans=cheng(ans,a);
    return ans;
}
int main()
{
    while (~scanf("%d",&n))
    {
        if (n==-1) return 0;
        if (!n)
        {
            printf("0\n");
            continue;
        }
        memset(st.a,0,sizeof(st.a));
        for (int i=1;i<=2;++i) st.a[i][i]=1;
        m.a[1][1]=m.a[1][2]=m.a[2][1]=1; m.a[2][2]=0;
        ans=matrix_fast_pow(m,n);
        printf("%d\n",ans.a[1][2]%Mod);
    }
}