Cod sursa(job #1893068)

Utilizator roxanastRoxana Stiuca roxanast Data 25 februarie 2017 14:27:50
Problema Al k-lea termen Fibonacci Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.9 kb
#include <fstream>
#include <cstring>
#define modulo 666013
using namespace std;
int n,i;
int mat[3][3],sol[3][3];
ifstream f("kfib.in");
ofstream g("kfib.out");

inline void inmult(int a[][3],int b[][3],int c[][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])%modulo;
}

inline void ridput(int p,int m[][3]){
    int c[3][3],aux[3][3],i;
    memcpy(c,mat,sizeof(mat));
    m[0][0]=m[1][1]=1;
    for(i=0;(1<<i)<=p;i++){
        if(p&(1<<i)){
            memset(aux,0,sizeof(aux));
            inmult(m,c,aux);
            memcpy(m,aux,sizeof(aux));
        }
        memset(aux,0,sizeof(aux));
        inmult(c,c,aux);
        memcpy(c,aux,sizeof(c));
    }
}
int main()
{
    f>>n;
    mat[0][0]=0,mat[0][1]=1,mat[1][0]=1,mat[1][1]=1;
    ridput(n-1,sol);
    g<<sol[1][1];
    return 0;
}