Cod sursa(job #2977404)

Utilizator mihaicrisanMihai Crisan mihaicrisan Data 11 februarie 2023 15:38:12
Problema Al k-lea termen Fibonacci Scor 60
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.14 kb
#include <fstream>

using namespace std;

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

const int MOD = 666013;

int k;
long long fib[2][2];

void multiply(long long mat1[][2], long long mat2[][2]) {
    long long rslt[2][2] = {0};
    for (long long i = 0; i < 2; i++)
        for (long long k = 0; k < 2; k++)
            for (long long j = 0; j < 2; j++)
                rslt[i][j] += (mat1[i][k] % MOD * mat2[k][j]) % MOD;

    for (long long i = 0; i < 2; i++)
        for (long long j = 0; j < 2; j++)
            mat1[i][j] = rslt[i][j];
}

void expo(long long n) {
    long long p[2][2] = {0};
    p[0][0] = p[1][1] = 1; //initializare matricea I2
    while (n) {
        if (n % 2 == 1)
            multiply(p, fib);
        multiply(fib, fib);
        n /= 2;
    }
    for (int i = 0; i < 2; i++)
        for (int j = 0; j < 2; j++)
            fib[i][j] = p[i][j];
}

int main() {
    fib[0][1] = fib[1][0] = fib[1][1] = 1; //initializare matrice fibonacci

    fin >> k;

    if (k == 0) {
        fout << 0;
        return 0;
    }

    expo(k-1);
    fout << fib[1][1];

    return 0;
}