POJ 3070 Fibonacci(矩阵快速幂)

#include <iostream>
#include <string.h>
using namespace std;
typedef long long ll;
int mod=10000;
const int N=3;
int n=2;
int aa[N][N],tmp[N][N],res[N][N];
void mul(int a[][N],int b[][N],int n)
{
    
    
    memset(tmp,0,sizeof(tmp));
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++)
    for(int k=1;k<=n;k++)
    {
    
    
        tmp[i][j]+=a[i][k]*b[k][j];
        tmp[i][j]%=mod;
    }
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++)
    {
    
    
        a[i][j]=tmp[i][j];
    }
}
void init()
{
    
    
    memset(res,0,sizeof(res));
    memset(aa,0,sizeof(aa));
    for(int i=1;i<=n;i++)res[i][i]=1;
    aa[1][1]=1;aa[1][2]=1;aa[2][1]=1;
    
}
void fsp(ll b)
{
    
    
    while(b)
    {
    
    
        if(b%2==1)mul(res,aa,n);
        b=b/2;
        mul(aa,aa,n);
    }
}
int main()
{
    
    
    int n;
    int f[100];
    f[0]=0;
    f[1]=1;
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    while(cin>>n && ~n)
    {
    
    
        init();
        fsp(n-1);
        if(n==0)cout<<f[0]<<endl;
        else if(n==1)cout<<f[1]<<endl;
        else cout<<res[1][1]<<endl;
    return 0;
}