Cod sursa(job #1035424)

Utilizator thewildnathNathan Wildenberg thewildnath Data 18 noiembrie 2013 16:08:11
Problema Al k-lea termen Fibonacci Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.96 kb
#include<stdio.h>
#include<string.h>
#define MOD 666013

int mat[3][3],sol[3][3],aux[3][3],c[3][3];


inline void inmult(int a[3][3],int b[3][3],int c[3][3])
{
    int i,j,k;
    for(i=0;i<2;++i)
        for(j=0;j<2;++j)
            for(k=0;k<2;++k)
                c[i][j]=(c[i][j]+1LL*a[i][k]*b[k][j])%MOD;
}


inline void pow_log(int pow,int v[3][3])
{
    int i;
    memcpy(c,mat,sizeof(mat));
    v[0][0]=v[1][1]=1;

    for(i=0;(1<<i)<=pow;++i)
    {
        if(pow&(1<<i))
        {
            memset(aux,0,sizeof(aux));
            inmult(v,c,aux);
            memcpy(v,aux,sizeof(aux));
        }
        memset(aux,0,sizeof(aux));
        inmult(c,c,aux);
        memcpy(c,aux,sizeof(c));
    }
}


int main()
{
    freopen("kfib.in","r",stdin);
    freopen("kfib.out","w",stdout);
    int k;
    scanf("%d",&k);

    mat[0][1]=mat[1][0]=mat[1][1]=1;
    pow_log(k-1,sol);

    printf("%d\n",sol[1][1]);

    return 0;
}