Cod sursa(job #1491957)

Utilizator andreiudilaUdila Andrei andreiudila Data 26 septembrie 2015 18:46:27
Problema Iepuri Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.47 kb
#include <fstream>
using namespace std;
ifstream fin("iepuri.in");
ofstream fout("iepuri.out");
struct matrice{
int a[4][4];

};

matrice m;
int t,x,y,z,a,b,c,mod=666013,i,j,k,n;

matrice matr(matrice a, matrice b)
{
        matrice rez;
        for(i=1;i<=3;i++)
            for(j=1;j<=3;j++)
                rez.a[i][j]=0;

        for(i=1;i<=3;i++)
            for(j=1;j<=3;j++)
                for(k=1;k<=3;k++)
                    rez.a[i][j]=rez.a[i][j]+a.a[i][k]*b.a[k][j];
    return rez;
}

matrice putere(matrice m, int n)
{
    matrice p;
    for(i=1;i<=3;i++)
        for(j=1;j<=3;j++)
            p.a[i][j]=0;
    for(i=1;i<=3;i++) p.a[i][i]=1;

    while(n>0)
    {
        if(n%2==0)
        {
        m=matr(m,m);
        n=n/2;
        }
        else
        {
            p=matr(p,m);
            n--;
        }


    }
 return p;
}
int main()
{
    int p=0;
    fin>>t;
    for(p=1;p<=t;p++)
    {
        fin>>x>>y>>z>>a>>b>>c>>n;

        m.a[1][1]=0;
        m.a[1][2]=0;
        m.a[1][3]=c;
        m.a[2][1]=1;
        m.a[2][2]=0;
        m.a[2][3]=b;
        m.a[3][1]=0;
        m.a[3][2]=1;
        m.a[3][3]=a;

        if(n==0) fout<<x<<'\n';
        else
        if(n==1) fout<<y<<'\n';
        else
        if(n==2) fout<<z<<'\n';
        else
        {
          m=putere(m, n-2);
          fout<<(x*m.a[1][3] + y*m.a[2][3] + z*m.a[3][3])%mod<<'\n';

        }


    }
    return 0;
}