POJ3070:Fibonacci(矩阵快速幂模板题)

时间:2021-11-12 11:16:32

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

#include <iostream>
#include <string.h>
#include <stdlib.h>
#include <cstdio>
#include <algorithm>
#define mod 10000
using namespace std;
struct m
{
int a[][];
} init,res;
int n;
m Mult(m x,m y)
{
m tmp;
for(int i=; i<; i++)
{
for(int j=; j<; j++)
{
tmp.a[i][j]=;
for(int k=; k<; k++)
{
tmp.a[i][j]=(tmp.a[i][j]+x.a[i][k]*y.a[k][j])%mod;
}
}
}
return tmp;
}
m Pow(m x,int n)
{
m tmp;
for(int i=; i<; i++)//将矩阵tmp初始化成单位矩阵[1,0]
{
for(int j=; j<; j++)// [0,1]
{
tmp.a[i][j]=(i==j);
}
}
while(n)
{
if(n&) tmp=Mult(tmp,x);
n>>=;
x=Mult(x,x);
}
return tmp;
}
int main()
{
while(scanf("%d",&n)!=EOF)
{
if(n==-) break;
init.a[][]=;
init.a[][]=;
init.a[][]=;
init.a[][]=;
res=Pow(init,n);
cout<<res.a[][]<<endl;
}
return ;
}