poj 3070 Fibonacci

矩阵好神奇~~

#include <cstdio>
#include <iostream>
using namespace std;
#define MOD 10000
#define MN 10
#define MM 10
int N;
struct matrix
{
       int a[MN][MM];
}origin,res,ans;
matrix multiply(matrix &x,matrix &y)
{
       matrix temp;
   //    memset(temp.a,0,sizeof(temp.a));
       for(int i=0;i<N;i++)
       {
           for(int j=0;j<N;j++)
            {
                temp.a[i][j]=0;
                for(int k=0;k<N;k++)
                {
                    temp.a[i][j]=(temp.a[i][j]+(x.a[i][k])*(y.a[k][j]));
                }
                temp.a[i][j]%=MOD;
            }
       }
       return temp;
}
void init()
{
     origin.a[0][0]=origin.a[0][1]=origin.a[1][0]=1;
     origin.a[1][1]=0;
     memset(res.a,0,sizeof(res.a));
     for(int i=0;i<N;i++)
        res.a[i][i]=1;
}

void calc(int n)
{
     while(n!=1)
     {
        if(n&1){n--;
        res=multiply(res,origin);}
        n>>=1;
        origin=multiply(origin,origin);
     }
     ans=multiply(res,origin);
}
int main()
{
    int n,cas;
    N=2;
    while(scanf("%d",&n)&&n!=-1)
    {
        if(!n) {puts("0");continue;}
        init();
        calc(n);
        printf("%d\n",ans.a[0][1]%MOD);
    }
    return 0;
}