Cod sursa(job #3123168)

Utilizator mariusn01Marius Nicoli mariusn01 Data 22 aprilie 2023 12:38:34
Problema Al k-lea termen Fibonacci Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.55 kb
#include <fstream>
#define MOD 666013
using namespace std;
ifstream fin("kfib.in");
ofstream fout("kfib.out");

void copiere(long long a[2][2], long long b[2][2]) {
    for (int i=0;i<=1;i++)
        for (int j=0;j<=1;j++)
            a[i][j] = b[i][j];
}

/// inmultirea a doua matrice 2X2 care ambele au liniile si coloanele
/// indexate de la 0
void inmultire(long long a[2][2], long long b[2][2], long long c[2][2]) {
    int i, j, k;
    for (i=0;i<=1;i++)
        for (j=0;j<=1;j++) {
            c[i][j] = 0;
            for (k=0;k<=1;k++) {
                c[i][j] += a[i][k] * b[k][j];
                c[i][j] %= MOD;
            }
        }
}

int n;

int main() {
    fin>>n;
    if (n<=2) {
        fout<<1;
        return 0;
    }

    n-=2;
    long long a[2][2] = { {1,1},
                          {1,0} };

    long long p[2][2] = { {1,0},
                          {0,1} };
    /// matricea p de mai sus este matricea element neutru de la inmultirea matricelor
    /// aceasta este matricea cu 1 pe diagonala principala si 0 in rest, numita
    /// si "matricea identitate" p * A = A  si A * p = A (asta e mereu adevarat la matricea identitate)
    /// p este un fel de "1" pentru inmultirea matricelor.

    long long b[2][2];
    while (n) {
        if (n%2 == 1) {
            inmultire(p, a, b); /// b = p*a
            copiere(p, b); /// p = b
        }
        inmultire(a, a, b);  /// b = a*a
        copiere(a, b);       /// a = b

        n/=2;

    }
    fout<<(p[0][0] + p[0][1])%MOD;
    return 0;
}