Cod sursa(job #2282891)

Utilizator mlupseLupse-Turpan Mircea mlupse Data 14 noiembrie 2018 18:20:03
Problema Al k-lea termen Fibonacci Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.84 kb
#include <fstream>
using namespace std;
ifstream fin("kfib.in");
ofstream fout("kfib.out");
const int MOD = 666013;

int K;
int Z[2][2],Sol[2][2];


void Multiply(int A[2][2], int B[2][2], int m, int n ,int p)
{
    int C[2][2];
    for(int i = 0; i < m; ++i)
        for(int j = 0; j < n; ++j)
            {
                C[i][j] = 0;
                for(int k = 0; k < p; ++k)
                    C[i][j] = (C[i][j] + (1LL * A[i][k]*B[k][j]))%MOD;

            }
    for(int i = 0; i < m; ++i)
        for(int j = 0; j < n; ++j)
            A[i][j] = C[i][j];
}

int main()
{
    Z[0][1] = Z[1][0] = Z[1][1] = 1;
    Sol[0][1] = 1;
    fin >> K;
    while(K)
    {
        if(K % 2 == 1)
            Multiply(Sol,Z,1,2,2);
        Multiply(Z,Z,2,2,2);
        K = K / 2;
    }
    fout << Sol[0][0];

    return 0;
}