Cod sursa(job #2593637)

Utilizator grecubogdanGrecu Bogdan grecubogdan Data 4 aprilie 2020 12:42:40
Problema Iepuri Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.76 kb
#include<fstream>
using namespace std;
const int MOD=666013;
long long mat[4][4],aux[4][4];
ifstream f("iepuri.in");
ofstream g("iepuri.out");
int i,j,k,a,b,c,x,y,z,n;
void solve(int a, int b, int c, int n)
{
    int i,j,k;
    n-=2;
    for(i=0;i<=3;i++)
        for(j=0;j<=3;j++)
            aux[i][j]=0;
    aux[0][0]=aux[1][1]=aux[2][2]=1;
    mat[0][0]=mat[0][1]=0,mat[0][2]=c;
    mat[1][0]=1,mat[1][1]=0,mat[1][2]=b;
    mat[2][0]=0,mat[2][1]=1,mat[2][2]=a;
    while(n)
        {
        if(n%2)
        {
            long long aux2[4][4];
            for(i=0;i<=3;i++)
                for(j=0;j<=3;j++)
                    aux2[i][j]=0;
            for(i=0;i<=2;i++)
                for(j=0;j<=2;j++)
                    for(k=0;k<=2;k++)
                        aux2[i][j]=(aux2[i][j]+aux[i][k]*mat[k][j])%MOD;
            for(i=0;i<=2;i++)
                for(j=0;j<=2;j++)
                    aux[i][j]=aux2[i][j];
        }
        n/=2;
        long long aux2[4][4];
        for(i=0;i<=3;i++)
            for(j=0;j<=3;j++)
                aux2[i][j]=0;
        for(i=0;i<=2;i++)
            for(j=0;j<=2;j++)
                for(k=0;k<=2;k++)
                    aux2[i][j]=(aux2[i][j]+mat[i][k]*mat[k][j])%MOD;
        for(i=0;i<=2;i++)
            for(j=0;j<=2;j++)
                mat[i][j]=aux2[i][j];
    }
}
int main()
{
    int t;
    f>>t;
    for(i=1;i<=t;i++)
        {
        long long v[3];
        f>>x>>y>>z>>a>>b>>c>>n;
        v[0]=x,v[1]=y,v[2]=z;
        solve(a, b, c, n);
        for(j=0;j<=2;j++)
            {
            long long s=0;
            for(k=0;k<=2;k++)
                s=(s+v[k]*aux[k][j])%MOD;
            if(j==2)
                g<<s%MOD<<'\n';
        }
    }
    return 0;
}