POJ 3070 矩阵快速幂

题目大意：求斐波那契数列数列的第n项的后四位，如果有前导0就不输出0

(斐波那契的另一个公式)

矩阵快速幂求斐波那契数列

#include <cstdio>
#include <cstring>
const int mod=10000;
struct matrix
{
    int m[2][2];
};

matrix mul(matrix a,matrix b)
{
    matrix tmp;
    memset(tmp.m,0,sizeof(tmp.m));
    for(int i=0;i<2;i++)
        for(int j=0;j<2;j++)
        {
            for(int k=0;k<2;k++)
                tmp.m[i][j]=tmp.m[i][j]+a.m[i][k]*b.m[k][j]%mod;
            tmp.m[i][j]%=mod;
        }
    return tmp;
}

int cal(int n)
{
    matrix a,b;
    a.m[0][0]=a.m[1][1]=1;
    a.m[1][0]=a.m[0][1]=0;
    b.m[0][0]=b.m[0][1]=b.m[1][0]=1;
    b.m[1][1]=0;
    while(n>0)
    {
        if(n&1)
            a=mul(a,b);
        b=mul(b,b);
        n>>=1;
    }
    return a.m[0][1];
}

int main()
{
    int n;
    while(scanf("%d",&n)&&n!=-1)
    {
        printf("%d
",cal(n));
    }
    return 0;
}