Cod sursa(job #3124318)

Utilizator AlexTimplaruAlexandru Timplaru AlexTimplaru Data 27 aprilie 2023 20:18:02
Problema Iepuri Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.52 kb
#include <fstream>
#define MOD 666013

using namespace std;

ifstream fin("iepuri.in");
ofstream fout("iepuri.out");

void inmultire(long long a[3][3], long long b[3][3], long long r[3][3])
{
    for(int i = 0; i < 3; i++)
    {
        for(int j = 0; j < 3; j++)
        {
            r[i][j] = 0;
            for(int k = 0; k < 3; k++)
            {
                r[i][j] += a[i][k] * b[k][j];
                r[i][j] %= MOD;
            }
        }
    }
}

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

void putere(long long a[3][3], long long n, long long r[3][3])
{
    long long i3[3][3] = { {1, 0, 0}, {0, 1, 0}, {0, 0, 1} };
    copiere(i3, r);

    long long t[3][3];
    while(n != 0)
    {
        if(n%2)
        {
            inmultire(r, a, t);
            copiere(t, r);
        }
        inmultire(a, a, t);
        copiere(t, a);
        n /= 2;
    }
}

int t;

int main()
{
    fin >> t;

    for(int i = 0; i < t; i++)
    {
        int x, y, z, a, b, c, n;
        fin >> x >> y >> z >> a >> b >> c >> n;
        long long ziua0[3][3] = { {z, y, x}, {0, 0, 0}, {0, 0, 0} };
        long long mat[3][3] = { {a, 1, 0}, {b, 0, 1}, {c, 0, 0} };

        long long p[3][3];
        putere(mat, n-2, p);

        long long r[3][3];
        inmultire(ziua0, p, r);

        fout << r[0][0] << "\n";
    }

    return 0;
}