Cod sursa(job #774779)

Utilizator tzipleatudTudor Tiplea tzipleatud Data 6 august 2012 14:54:52
Problema Iepuri Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.58 kb
#include <fstream>
#include <cstring>
#define MOD 666013

using namespace std;

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


int A,B,C,X,Y,Z,N;
long long M1[3][3],MX[3][3],AX[3][3],R[3][3];

void Multiply (long long A[][3], long long B[][3], long long C[][3])
{
    int i,j,k;
    for (i=0; i<3; i++)
        for (j=0; j<3; j++)
            C[i][j]=0;
    for (i=0; i<3; i++)
        for (j=0; j<3; j++)
            for (k=0; k<3; k++)
            {
                C[i][j]=(C[i][j]+A[i][k]*B[k][j]*1LL);
                if (C[i][j]>=MOD) C[i][j]%=MOD;
            }
}

void LgPow (int P)
{
    for (int i=0; i<3; i++)
        for (int j=0; j<3; j++)
            R[i][j]=0;
    R[0][0]=R[1][1]=R[2][2]=1;
    for (int i=1; i<=P; i<<=1)
    {
        if (i&P)
        {
            Multiply(R,MX,AX);
            memcpy(R,AX,sizeof(R));
        }
        Multiply(MX,MX,AX);
        memcpy(MX,AX,sizeof(MX));
    }
    memcpy(MX,R,sizeof(MX));
}

int Solve ()
{
    M1[0][0]=Z;
    M1[0][1]=Y;
    M1[0][2]=X;
    M1[1][0]=0;
    M1[1][1]=0;
    M1[1][2]=0;
    M1[2][0]=0;
    M1[2][1]=0;
    M1[2][2]=0;
    MX[0][0]=A;
    MX[0][1]=1;
    MX[0][2]=0;
    MX[1][0]=B;
    MX[1][1]=0;
    MX[1][2]=1;
    MX[2][0]=C;
    MX[2][1]=0;
    MX[2][2]=0;
    LgPow(N);
    Multiply(M1,MX,AX);
    return AX[0][0];
}


int T,n,x,y,z,a,b,c;

int main ()
{
    for (f >> T; T; --T)
    {
        f >> X >> Y >> Z >> A >> B >> C >> N;
        N-=2;
        g << Solve() << '\n';
    }
    f.close();
    g.close();
    return 0;
}