poj -3070 Fibonacci (矩阵快速幂)

http://poj.org/problem?id=3070

直接用矩阵快速幂得到f(n)

然后输出后四位即可。

#include <cstdio>
typedef long long ll;
const int mod = 10000;
struct Mat
{
    ll matrix[2][2];
};

Mat mul(Mat a,Mat b)
{
    Mat c;
    for(int i=0;i<2;i++)
        for(int j=0;j<2;j++)
        {
            c.matrix[i][j]=0;
            for(int k=0;k<2;k++)
            {
                c.matrix[i][j]+=(a.matrix[i][k]*b.matrix[k][j])%mod;
            }
            c.matrix[i][j]%=mod;
        }
    return c;
}
Mat mat;
Mat solve(ll m)
{
    Mat mt=mat;
    m--;
    while(m)
    {
        if(m&1)
        {
            mt=mul(mat,mt);
            m--;
        }
        mat=mul(mat,mat);
        m/=2;
    }
    return mt;
}
int main()
{
    //freopen("a.txt","r",stdin);
    ll n;
    while(~scanf("%lld",&n)&&n!=-1)
    {
        if(n==0) {printf("0
");continue;}
        if(n==1) {printf("1
");continue;}
        mat.matrix[0][0]=mat.matrix[0][1]=mat.matrix[1][0]=1;
        mat.matrix[1][1]=0;
        Mat c=solve(n-1);
        printf("%lld
",c.matrix[0][0]%10000);
    }
    return 0;
}