Cod sursa(job #1788613)

Utilizator OlivianOlivian Dan Cretu Olivian Data 26 octombrie 2016 10:21:49
Problema Al k-lea termen Fibonacci Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.03 kb
#include <cstdio>
#include <vector>

using namespace std;

const int mod=666013;
int dim;

struct matrice
{
    vector<vector<int> > v;
    matrice()
    {
        v.resize(dim+1,vector<int>(dim+1,0));
    }
    void unitate()
    {
        for(int i=1;i<=dim;i++) v[i][i]=1;
    }
    matrice operator *(const matrice &aux) const
    {
        matrice ret;
        for(int i=1;i<=dim;i++)
            for(int j=1;j<=dim;j++)
                for(int k=1;k<=dim;k++)
                    ret.v[i][j]=(ret.v[i][j]+1LL*v[i][k]*aux.v[k][j])%mod;
        return ret;
    }
};

matrice rid_put(matrice a,int n)
{
    matrice p;
    p.unitate();
    for(int i=1;i<=n;i<<=1)
    {
        if(n&i) p=p*a;
        a=a*a;
    }
    return p;
}

int main()
{
    freopen("kfib.in", "r", stdin);
    freopen("kfib.out", "w", stdout);
    int k;
    scanf("%d",&k);
    dim=2;
    matrice a;
    a.v[1][1]=0;
    a.v[1][2]=a.v[2][1]=a.v[2][2]=1;
    a=rid_put(a,k-1);
    printf("%d",a.v[2][2]);
    return 0;
}