Cod sursa(job #2691282)

Utilizator emma.chirlomezEmma Chirlomez emma.chirlomez Data 27 decembrie 2020 21:03:33
Problema Al k-lea termen Fibonacci Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.83 kb
#include <bits/stdc++.h>
using namespace std;

const int MOD = 666013;

struct Mat
{
    int v[2][2];
    Mat(){
        v[0][0] = v[0][1] = v[1][0] = v[1][1] = 0;
    }
};

Mat operator*(Mat a, Mat b)
{
    Mat ans;
    for(int i = 0; i <= 1; i++)
        for(int j = 0; j <= 1; j++)
            for(int k = 0; k <= 1; k++)
                ans.v[i][j] = (ans.v[i][j] + 1LL * a.v[i][k] * b.v[k][j]) % MOD;
    return ans;
}

Mat Put(Mat x, int p)
{
    if (p == 0) {
        Mat ans;
        ans.v[0][0] = ans.v[1][1] = 1;
        return ans;
    }
    if (p % 2 == 1)
        return x * Put(x, p - 1);
    return Put(x * x, p / 2);
}

int main()
{
    ifstream in("kfib.in");
    ofstream out("kfib.out");

    int k;
    in >> k;

    Mat x;
    x.v[0][0] = x.v[1][0] = x.v[0][1] = 1;

    x = Put(x, k);
    out << x.v[1][0] << '\n';
    return 0;
}