Cod sursa(job #2943723)

Utilizator rapidu36Victor Manz rapidu36 Data 21 noiembrie 2022 16:14:26
Problema Al k-lea termen Fibonacci Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.16 kb
#include <fstream>

using namespace std;

const int M = 666013;
const int N = 2;

void produs(int p[N][N], int a[N][N])
{
    int aux[N][N];
    for (int i = 0; i < N; i++)
    {
        for (int j = 0; j < N; j++)
        {
            ///calculez aux[i][j] = produs scalar intre linia i din p si coloana j din a
            aux[i][j] = 0;
            for (int k = 0; k < N; k++)
            {
                aux[i][j] += (long long)p[i][k] * a[k][j] % M;
                aux[i][j] %= M;
            }
        }
    }
    for (int i = 0; i < N; i++)
    {
        for (int j = 0; j < N; j++)
        {
            p[i][j] = aux[i][j];
        }
    }
}

int fibo(int n)
{
    int a[2][2] = {{1, 1}, {1, 0}};
    int p[2][2] = {{1, 0}, {0, 1}};
    n--;
    while (n != 0)
    {
        int cifb = n % 2;
        if (cifb != 0)
        {
            produs(p, a);///p *= a pentru matrice
        }
        produs(a, a);
        n /= 2;
    }
    return p[0][0];
}

int main()
{
    ifstream in("kfib.in");
    ofstream out("kfib.out");
    int k;
    in >> k;
    out << fibo(k);
    in.close();
    out.close();
    return 0;
}