Cod sursa(job #2716995)

Utilizator AlexMariMarinescu Alexandru AlexMari Data 6 martie 2021 08:19:01
Problema Iepuri Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.47 kb
#include<bits/stdc++.h>
using namespace std;
ifstream fin("iepuri.in");
ofstream fout("iepuri.out");

const long long MOD=666013;

long long X,Y,Z,A,B,C,n;

long long sol[5][5],a[5][5],ans[5][5],matrice_de_start[5][5],c[5][5],d[5][5];

void inmultire_matrice(long long c[5][5],long long d[5][5])
{
    for(int i=1;i<=3;i++)
        for(int j=1;j<=3;j++)
    {
        long long s=0;
        for(int k=1;k<=3;k++)
            {
                s+=c[i][k]*d[k][j];
                s=s%MOD;
            }
        sol[i][j]=s;
    }
    for(int i=1;i<=3;i++)
        for(int j=1;j<=3;j++)
          c[i][j]=sol[i][j];
}

void lgput(long long k)
{
    while(k)
    {
        if(k%2==1)
            inmultire_matrice(ans,a);
        inmultire_matrice(a,a);
        k/=2;
    }
}


int main()
{
    int t;
    fin>>t;
    while(t--)
    {
        for(int i=1;i<=3;i++)
            for(int j=1;j<=3;j++)
        {
            ans[i][j]=0;
            a[i][j]=0;
            matrice_de_start[i][j]=0;
        }
        fin>>X>>Y>>Z>>A>>B>>C>>n;
        matrice_de_start[1][1]=X;
        matrice_de_start[1][2]=Y;
        matrice_de_start[1][3]=Z;
        a[1][3]=C;
        a[2][3]=B;
        a[3][3]=A;
        a[2][1]=1;
        a[3][2]=1;
        ans[1][1]=1;
        ans[2][2]=1;
        ans[3][3]=1;
        lgput(n-2);
        inmultire_matrice(matrice_de_start,ans);
        fout<<matrice_de_start[1][3]<<'\n';
    }
    return 0;
}