Cod sursa(job #1558021)

Utilizator drobertDumitru Robert drobert Data 28 decembrie 2015 16:48:21
Problema Iepuri Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.64 kb
#include <fstream>
#include <string.h>
using namespace std;
ifstream f("iepuri.in");
ofstream g("iepuri.out");

#define mod 666013

int tests, x, y, z, a, b, c, n;
int m[5][5], t[5][5], sol[5][5];

void inm1()
{
    int i, j, k, r[5][5];
    for (i = 1;i <= 3;i++)
        for (j = 1;j <= 3;j++)
            r[i][j] = 0;
    for (i = 1;i <= 3;i++)
        for (j = 1;j <= 3;j++)
        {
            for (k = 1;k <= 3;k++)
                r[i][j] += (1LL * sol[i][k] * m[k][j]) % mod;
            r[i][j] %= mod;
        }
    memcpy(sol, r, sizeof(r));
}

void inm2()
{
    int i, j, k, r[5][5];
    for (i = 1;i <= 3;i++)
        for (j = 1;j <= 3;j++)
            r[i][j] = 0;
    for (i = 1;i <= 3;i++)
        for (j = 1;j <= 3;j++)
        {
            for (k = 1;k <= 3;k++)
                r[i][j] += (1LL * m[i][k] * m[k][j]) % mod;
            r[i][j] %= mod;
        }
    memcpy(m, r, sizeof(r));
}

void expo(long long p)
{
    long long i, j;
    for (i = 0;(1<<i) <= p;i++)
    {
        if (((1<<i) & p) > 0)
            inm1();
        inm2();
    }
}

int main()
{
    int i, j, temp, t1, t2, t3;
    for (f>>tests;tests;tests--)
    {
        f>>x>>y>>z>>a>>b>>c>>n;
        m[1][1] = 0; m[1][2] = 1; m[1][3] = 0;
        m[2][1] = 0; m[2][2] = 0; m[2][3] = 1;
        m[3][1] = c; m[3][2] = b; m[3][3] = a;
        memcpy(sol, m, sizeof(m));
        expo(n - 3);
        c = sol[3][1]; b = sol[3][2]; a = sol[3][3];
        t1 = (1LL * c * x) % mod;
        t2 = (1LL * b * y) % mod;
        t3 = (1LL * a * z) % mod;
        temp = (t1 + t2 + t3) % mod;
        g<<temp<<'\n';
    }
}