Cod sursa(job #1643372)

Utilizator gbibBacotiu Gabi gbib Data 9 martie 2016 18:39:48
Problema Al k-lea termen Fibonacci Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.06 kb
#include <bits/stdc++.h>
#define mod 666013
using namespace std;
ifstream in("kfib.in");
ofstream out("kfib.out");
void inmult(long long a[ ][3], long long b[ ][3]) //a<-a*b
{
    long long c[3][3];
    c[0][0]=0;
    c[0][1]=0;
    c[1][0]=0;
    c[1][1]=0;
    int i,k,j;
    for(i=0;i<=1;i++)
    {
        for(j=0;j<=1;j++)
        {
            for(k=0;k<=1;k++)
            {
                c[i][j]=(c[i][j]+a[i][k]*b[k][j])%mod;
            }
        }
    }
    a[0][0]=c[0][0];
    a[0][1]=c[0][1];
    a[1][0]=c[1][0];
    a[1][1]=c[1][1];
}

void put(long long a[][3],long long sol[][3], long long p)
{
    long long bit;
    for(bit=0;(1<<bit)<=p;bit++)
    {
        if((1<<bit)&p)
        {
            inmult(sol,a);
        }
        inmult(a,a);
    }
}


int main()
{long long p;
long long a[3][3];
a[1][0]=a[0][1]=a[0][0]=1;
a[1][1]=0;
long long sol[3][3];
sol[1][1]=sol[0][0]=1;
sol[1][0]=sol[0][1]=0;
in>>p;
if(p==0)
{
    out<<0<<'\n';
    return 0;
}
put(a,sol,p-1);
out<<sol[0][0]<<" ";
    return 0;
}