Cod sursa(job #2574530)

Utilizator Fatu_SamuelFatu Samuel Fatu_Samuel Data 5 martie 2020 23:25:23
Problema Al k-lea termen Fibonacci Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.04 kb
#include <bits/stdc++.h>

using namespace std;

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

typedef vector < vector < int > > Matrix;

const int MOD = 666013;

int n;

Matrix aux, r;

void Resize(Matrix &a, int s)
{
    a.resize(s);

    for (int i = 0; i < s; i++)
        a[i].resize(s);
}

Matrix Inmultire(Matrix a, Matrix b)
{
    Matrix c;

    Resize(c, 2);

    for (int i = 0; i < a.size(); i++)
        for (int j = 0; j < a.size(); j++)
            for (int k = 0; k < a.size(); k++)
                c[i][j] += (1LL * a[i][k] * b[k][j]) % MOD;

    return c;
}

int main()
{
    fin >> n;

    n--;

    Resize(r, 2);
    Resize(aux, 2);

    aux[0][0] = aux[1][1] = 1;
    r[0][1] = r[1][0] = r[1][1] = 1;

    while (n != 1)
    {
        if (n % 2)
        {
            aux = Inmultire(aux, r);
        }

        r = Inmultire(r, r);


        n /= 2;
    }

    r = Inmultire(r, aux);

    fout << r[1][1] % MOD;

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