Cod sursa(job #2550176)

Utilizator FrostfireMagirescu Tudor Frostfire Data 18 februarie 2020 15:46:22
Problema Iepuri Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.41 kb
#include <iostream>
#include <fstream>
#define MOD 666013

using namespace std;

ifstream f("iepuri.in");
ofstream g("iepuri.out");

struct mat
{   long long m[5][5];
}A, ans;

int t;

void init(long long x, long long y, long long z, long long a, long long b, long long c)
{   for(int i=0; i<3; i++)
        for(int j=0; j<3; j++) A.m[i][j] = 0, ans.m[i][j] = 0;
    A.m[1][0] = A.m[2][1] = 1;
    A.m[0][2] = c;
    A.m[1][2] = b;
    A.m[2][2] = a;
    ans.m[0][0] = x;
    ans.m[0][1] = y;
    ans.m[0][2] = z;
}

mat prod(mat A, mat B)
{   mat sol;
    for(int i=0; i<3; i++)
        for(int j=0; j<3; j++) sol.m[i][j] = 0;
    for(int i=0; i<3; i++)
        for(int j=0; j<3; j++)
            for(int k=0; k<3; k++) sol.m[i][j] = (sol.m[i][j] + A.m[i][k] * B.m[k][j]) % MOD;
    return sol;
}

void lgput(long long n)
{   while(n)
        {   if(n % 2 == 0) A = prod(A, A), n /= 2;
            else ans = prod(ans, A), n--;
        }
}

int main()
{
    f >> t;
    while(t--)
        {   long long x, y, z, a, b, c, n;
            f >> x >> y >> z >> a >> b >> c >> n;
            if(!n) g << x << '\n';
            else if(n == 1) g << y << '\n';
            else if(n == 2) g << z << '\n';
            else
                {   init(x, y, z, a, b, c);
                    lgput(n-2);
                    g << ans.m[0][2] << '\n';
                }
        }
    return 0;
}