Cod sursa(job #2217427)

Utilizator Andrei-27Arhire Andrei Andrei-27 Data 30 iunie 2018 13:44:20
Problema Iepuri Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.65 kb
#include <iostream>
using namespace std;
#include <fstream>
long long a[3][3],n;
ifstream f("iepuri.in");
ofstream g("iepuri.out");
int X,Y,Z,A,B,C;

unsigned pow (long long a[3][3],long long n)
{
    unsigned i,j,k;
     if (n==1)
        return 0;
    long long c[3][3]={0},u[3][3]={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]*a[k][j])%666013;
        for(i=0;i<3;i++)
            for(j=0;j<3;j++)
                {u[i][j]=a[i][j];
                a[i][j]=c[i][j];}



    if(n%2==0)
        pow(a,n/2);
    if(n%2==1)
    {   pow(a,n/2);
    long long d[3][3]={0};
         for(i=0;i<3;i++)
            for(j=0;j<3;j++)
                for(k=0;k<3;k++)
                    d[i][j]=(d[i][j]+u[i][k]*a[k][j])%666013;
        for(i=0;i<3;i++)
        for(j=0;j<3;j++)
        a[i][j]=d[i][j];}
}

void simplu (long long a[3][3], long long s[3][1])
{
    unsigned i,k;
    long long x[3][1]={0};
    for(i=0;i<3;i++)
        for(k=0;k<3;k++)
            x[i][0]=(x[i][0]+s[k][0]*a[i][k])%666013;
    g<<x[0][0]<<"\n";
}


int main ()
{
int t;
f>>t;
long long s[3][1];

while (t>0)
{
    f>>X>>Y>>Z>>A>>B>>C>>n;

a[0][0]=A,a[0][1]=B,a[0][2]=C;
a[1][0]=1,a[1][1]=0,a[1][2]=0,a[2][0]=0,a[2][1]=1,a[2][2]=0;
s[0][0]=Z,s[1][0]=Y,s[2][0]=X;
    if (n==0)
        {
        g<<X;
        return 0;
        }
    if(n==1)
    {
        g<<Y;
        return 0;
    }
     if(n==2)
    {
        g<<Z;
        return 0;
    }
        n-=2;
     pow(a,n);
    simplu(a,s);
   t--;
}

    return 0;
}