poj3070

矩阵+二分

练手

#include <stdio.h>
#include <string.h>
#define MAXN 2
#define MOD(a) ((a)%10000)
void MatrixMul(int max0[][MAXN],int max1[][MAXN],int max2[][MAXN])
{
    for(int i=0;i<2;++i)
        for(int j=0;j<2;++j)
        {
            for(int k=0;k<2;++k)
                max2[i][j]+=MOD(max0[i][k]*max1[k][j]);
            max2[i][j]=MOD(max2[i][j]);
        }
}
void MatrixCpy(int max0[][MAXN],int max1[][MAXN])
{
    for(int i=0;i<2;++i)
        for(int j=0;j<2;++j)
            max1[i][j]=max0[i][j];
}
int main()
{
    int matrix[3][MAXN][MAXN];
    int n;
    while(scanf("%d",&n),n!=-1)
    {
        if(n==0)
        {
            printf("0\n");
            continue;
        }
        matrix[0][0][0]=matrix[0][1][1]=1;matrix[0][0][1]=matrix[0][1][0]=0;
        matrix[1][0][0]=matrix[1][0][1]=matrix[1][1][0]=1;matrix[1][1][1]=0;
        --n;
        while(n>0)
        {
            if(n&1)
            {
                memset(matrix[2],0,sizeof(matrix[2]));
                MatrixMul(matrix[0],matrix[1],matrix[2]);
                MatrixCpy(matrix[2],matrix[0]);
            }
            n>>=1;
            memset(matrix[2],0,sizeof(matrix[2]));
            MatrixMul(matrix[1],matrix[1],matrix[2]);
            MatrixCpy(matrix[2],matrix[1]);
        }
        printf("%d\n",matrix[0][0][0]);
    }
    return 0;
}