Cod sursa(job #2227772)

Utilizator vladm98Munteanu Vlad vladm98 Data 1 august 2018 17:56:57
Problema Al k-lea termen Fibonacci Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.1 kb
#include <bits/stdc++.h>

using namespace std;

int MOD = 666013;

void mulMatrix (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 i = 0; i < 2; ++i) {
        for (int j = 0; j < 2; ++j) {
            for (int k = 0; k < 2; ++k) {
                c[i][j] = static_cast<int>((c[i][j] + 1LL * a[i][k] * b[k][j]) % MOD);
            }
        }
    }
    for (int i = 0; i < 2; ++i) {
        for (int j = 0; j < 2; ++j) {
            a[i][j] = c[i][j];
        }
    }
}

long long answer (long long k) {
    int answer[2][2];
    answer[0][0] = answer[1][1] = 1;
    answer[0][1] = answer[1][0] = 0;
    int base[2][2];
    base[0][0] = 0;
    base[0][1] = base[1][0] = base[1][1] = 1;

    while (k) {
        if (k % 2 == 1) {
            mulMatrix(answer, base);
        }
        mulMatrix(base, base);
        k /= 2;
    }
    return answer[1][0];
}

int main()
{
    ifstream fin ("kfib.in");
    ofstream fout ("kfib.out");
    long long k;
    fin >> k;
    fout << answer(k);
    return 0;
}