Cod sursa(job #2133750)

Utilizator crastanRavariu Eugen crastan Data 17 februarie 2018 11:27:50
Problema Iepuri Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.76 kb
#include <iostream>
#include <fstream>
using namespace std;
#define mod 666013
ifstream fin("iepuri.in");
ofstream fout("iepuri.out");
int t,x,y,z,a,b,c,n,i,yy,zz;
int matrice[4][4];
int neutru[4][4]={{0,0,0,0},{0,1,0,0},{0,0,1,0},{0,0,0,1}};
int finna[4][4],iep[4][4],term[4][4];
void multMatrici (int a[4][4],int b[4][4],int rez[4][4])
{
    for(int i=1;i<=3;i++)
        for(int j=1;j<=3;j++)
        {
            rez[i][j]=0;
            for(int k=1;k<=3;k++)
            {
               rez[i][j]+=(a[i][k]*b[k][j]%mod);
               rez[i][j]%=mod;


            }
        }

}
void atrib(int b[4][4],int a[4][4])
{
    //a->b
    for(int i=1;i<=3;i++)
        for(int j=1;j<=3;j++)
            b[i][j]=a[i][j];
}
void logPut(int baza[4][4],int exp,int rezult[4][4])
{
    int a[4][4];
    if(exp==1)
    {
        for(int i=1;i<=3;i++)
            for(int j=1;j<=3;j++)
            {
                atrib(rezult,baza);
            }
    }
    else
        if(n%2==0)
        {
            logPut(baza,exp/2,a);
            multMatrici(a,a,rezult);
        }
        else
        {int b[4][4];
            logPut(baza,exp/2,a);
            multMatrici(a,a,b);
            multMatrici(b,baza,rezult);
        }
}
int main()
{
    fin>>t;
    matrice[2][1]=matrice[3][2]=1;
    for(i=1;i<=t;i++)
    {
        fin>>x>>y>>z>>a>>b>>c>>n;
        iep[1][1]=x;
        iep[1][2]=y;
        iep[1][3]=z;
        matrice[1][3]=c;
        matrice[2][3]=b;
        matrice[3][3]=a;
        if(n==0) fout<<x;
        else if(n==1) fout<<x+y;
        else if(n==2) fout<<x+y+z;
        else
            logPut(matrice,n-2,finna);
        multMatrici(iep,finna,term);
        fout<<term[1][3]<<'\n';


    }
    return 0;
}