Cod sursa(job #2111719)

Utilizator RazorBestPricop Razvan Marius RazorBest Data 22 ianuarie 2018 16:53:05
Problema Al k-lea termen Fibonacci Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.22 kb
#include <fstream>
#include <iostream>
#define MOD 666013
using namespace std;

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

template <class T>
void copy_matrix(T a[2][2], T b[2][2])
{
    for (int i = 0; i < 2; i++)
        for (int j = 0; j < 2; j++)
            a[i][j] = b[i][j];
}

void mult(int a[2][2], int b[2][2], int c[2][2])
{
    for (int i = 0; i < 2; i++)
        for (int j = 0; j < 2; j++)
        {
            c[i][j] = 0;
            for (int k = 0; k < 2; k++)
            {
                long long p = (long long)a[i][k] * b[k][j];
                if (p >= MOD) p %= MOD;
                c[i][j] += p;
                if (c[i][j] >= MOD) c[i][j] %= MOD;
            }
        }
}

void exp_log(int a[2][2], int e, int b[2][2])
{
    if (e < 1)
        return;
    if (e == 1)
    {
        copy_matrix(b, a);
        return;
    }

    exp_log(a, e / 2, b);

    int c[2][2];
    mult(b, b, c);
    copy_matrix(b, c);

    if (e % 2 == 1)
    {
        mult(a, b, c);
        copy_matrix(b, c);
    }
}

int main()
{
    int k, a[2][2] = {{0, 1}, {1, 1}}, c[2][2] = {{1, 0}, {0, 1}};

    fin >> k;

    exp_log(a, k - 1, c);

    fout << c[1][1] % MOD;
}