poj 3070 Fibonacci(矩阵优化递推入门)

poj 3070 Fibonacci

题目给出了递推矩阵

#include<stdio.h>
#include<string.h>
#define MOD 10000
struct matrix
{
int m[3][3];
};
matrix multi(const matrix &a,const matrix &b)
{
matrix ans;
memset(&ans,0,sizeof(ans));
for(int i=0;i<2;i++)
for(int j=0;j<2;j++)
for(int k=0;k<2;k++)
{
ans.m[i][j]+=a.m[i][k]*b.m[k][j];
ans.m[i][j]%=MOD;
}
return ans;
}
matrix pow(matrix a,int k)
{
matrix ans;
memset(&ans,0,sizeof(ans));
for(int i=0;i<2;i++) ans.m[i][i]=1;
while(k)
{
if(k&1) ans=multi(ans, a);
a=multi(a, a);
k>>=1;
}
return ans;
}
int main()
{
matrix a;
int n;
while(scanf("%d",&n)!=EOF&&(n>=0))
{
memset(&a,0,sizeof(a));
a.m[0][0]=a.m[0][1]=a.m[1][0]=1;
a=pow(a,n);
printf("%d\n",a.m[0][1]);
}
return 0;
}