poj 3070

矩阵加速

#include <iostream>

#include <cstdio>
#include <cstring>
using namespace std;
int c[101][3][3],b[3][3],d[3][3],lon;
int n;


void dfs(int k)
{


    if(k==1)
    {
        lon++;
        for(int i=1;i<=2;i++)
        for(int j=1;j<=2;j++)
        c[lon][i][j]=b[i][j];
        return;
    }
    dfs(k/2);
    memset(d,0,sizeof(d));
    for(int p=1;p<=2;p++)
    for(int q=1;q<=2;q++)
    for(int i=1;i<=2;i++)
    {
        d[p][q]+=c[lon][p][i]*c[lon][i][q];
        d[p][q]%=10000;
    }
    lon++;
    if(k&1)
    {
        for(int p=1;p<=2;p++)
        for(int q=1;q<=2;q++)
        for(int i=1;i<=2;i++)
        {
            c[lon][p][q]+=d[p][i]*b[i][q];
            c[lon][p][q]%=10000;
        }
    }
    else
    {
        for(int p=1;p<=2;p++)
        for(int q=1;q<=2;q++)
        c[lon][p][q]=d[p][q];
    }
}


int main()
{
    while(scanf("%d",&n),n!=-1)
    {
        if(n<=2)
        {
            if(n==0) printf("0\n");
            else if(n==1) printf("1\n");
            else printf("2\n");
            continue;
        }
        b[1][1]=1;
        b[1][2]=1;
        b[2][1]=1;
        b[2][2]=0;
        memset(c,0,sizeof(c));
        lon=-1;
        dfs(n-2);
        int ans=(c[lon][1][1]+c[lon][2][1])%10000;
        printf("%d\n",ans);
    }
    return 0;
}