Cod sursa(job #2133760)

Utilizator natrovanCeval Marius natrovan Data 17 februarie 2018 11:44:26
Problema Iepuri Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.94 kb
#include <iostream>
#include <fstream>

using namespace std;
typedef unsigned long long mat[3][3];
#define MOD 666013

ifstream fin ("kfib.in");
ofstream fout ("kfib.out");

void inm (mat a, mat b, mat c)
{
    mat d;
    int i,j,k;
    for(i=0;i<3;++i)
        for(j=0;j<3;++j)
    {
        d[i][j]=0;
        for(k=0;k<3;++k)
            d[i][j]=a[i][k]*b[k][j]%MOD;
    }
}


void Alan(mat a,int n, mat b)
{
    if(n==0){b[0][0]=b[1][1]=b[2][2]=1;
             b[0][1]=b[1][0]=b[0][2]=b[1][2]=b[2][0]=b[2][1];}
    else{mat aux;
         Alan(a,n/2,aux);
         inm(aux,aux,b);
         if(n%2)inm(b,a,b);
        }
}

int k;
int main()
{
    int C,B,A;
    fin>>k;
    mat a,b;
    a[0][0]=0;
    a[0][1]=0;
    a[0][2]=C;
    a[1][0]=1;
    a[1][1]=0;
    a[1][2]=B;
    a[2][0]=0;
    a[2][1]=1;
    a[2][2]=A;
    Alan(a,k-1,b);
    fout<<b[1][1];


    fin.close(); fout.close();
    return 0;
}