Cod sursa(job #2133785)

Utilizator crastanRavariu Eugen crastan Data 17 februarie 2018 11:59:35
Problema Iepuri Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.42 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 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 multMatrici (int a[4][4],int b[4][4],int rez[4][4])
{ int aux[4][4];
    for(int i=1;i<=3;i++)
        for(int j=1;j<=3;j++)
        {
            aux[i][j]=0;
            for(int k=1;k<=3;k++)
            {
               aux[i][j]+=(a[i][k]*b[k][j]%mod);
               aux[i][j]%=mod;


            }
        }
        atrib(rez,aux);


}

void afis(int a[4][4])
{
    for(int i=1;i<=3;i++)
    {
        for(int j=1;j<=3;j++)
            cout<<a[i][j]<<" ";
        cout<<endl;
    }
    cout<<endl;
}
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(exp%2==0)
        {
            logPut(baza,exp/2,a);
            //cout<<"a"<<endl;
            //afis(a);

            multMatrici(a,a,rezult);
            //cout<<"rez"<<endl;
            //afis(rezult);
            //cout<<endl;
        }
        else
        {int b[4][4];
            logPut(baza,exp/2,a);
            //cout<<"a"<<endl;
            //afis(a);
            multMatrici(a,a,b);
            //cout<<"b"<<endl;
            //afis(b);
            multMatrici(b,baza,rezult);
            //cout<<"rez"<<endl;
            //afis(rezult);
            //cout<<endl;
        }
}
int main()
{
    fin>>t;
    matrice[2][1]=matrice[3][2]=1;
    for(i=1;i<=t;i++)
    {
        //cout<<endl;
        //cout<<11111111<<endl;
        //cout<<endl;
        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);
            //afis(finna);
        multMatrici(iep,finna,term);
        //afis(term);
        fout<<term[1][3]<<'\n';


    }
    return 0;
}