Cod sursa(job #2714670)

Utilizator divianegoescuDivia Negoescu divianegoescu Data 2 martie 2021 11:06:58
Problema Iepuri Scor 60
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.2 kb
#include <fstream>
#define R 666013
using namespace std;
ifstream fin("iepuri.in");
ofstream fout("iepuri.out");
int a[4][4],aux[4][4],sol[4][4],t,x,y,z,A,B,C,n,i,j,k;
void inmulteste(int a[4][4],int b[4][4],int c[4][4]){
    for(i=1;i<=3;i++)
        for(j=1;j<=3;j++){
            c[i][j]=0;
            for(k=1;k<=3;k++){
                c[i][j]+=1LL*a[i][k]*b[k][j]%R;
                c[i][j]%=R;
            }
    }
}
void init(int a[4][4],int b[4][4]){
    for(i=1;i<=3;i++)
        for(j=1;j<=3;j++)
            a[i][j]=b[i][j]=0;
}
void copiaza(int a[4][4],int b[4][4]){
    for(i=1;i<=3;i++)
        for(j=1;j<=3;j++)
            b[i][j]=a[i][j];
}
int main(){
    for(fin>>t;t--;){
        fin>>x>>y>>z>>A>>B>>C>>n;
        init(a,sol);
        sol[1][1]=sol[2][2]=sol[3][3]=1;
        a[1][1]=A;  a[1][2]=1;
        a[2][1]=B;  a[2][3]=1;
        a[3][1]=C;
        n-=2;
        while(n){
            if(n%2){
                inmulteste(sol,a,aux);
                copiaza(aux,sol);
            }
            inmulteste(a,a,aux);
            copiaza(aux,a);
            n/=2;
        }
        fout<<(1LL*z*sol[1][1]+y*sol[2][1]+x*sol[3][1])%R<<"\n";
    }
    return 0;
}