Cod sursa(job #2392703)

Utilizator darisavuSavu Daria darisavu Data 30 martie 2019 12:10:01
Problema Iepuri Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.51 kb
#include <fstream>
#define mod  666013
using namespace std;
ifstream f("iepuri.in");
ofstream g("iepuri.out");
int n;
long long sol[3][3],a[3][3],c[3][3];
void inmultire(long long A[3][3], long long B[3][3])
{
    long long C[3][3],i,j,k;
    for(i=0;i<=2;i++)
        for(j=0;j<=2;j++) C[i][j]=0;
    for(i=0;i<=2;i++)
        for(j=0;j<=2;j++)
    {
        for(k=0;k<=2;k++) C[i][j]=(C[i][j]+(A[i][k]*B[k][j])%mod)%mod;
    }
     for(i=0;i<=2;i++)
        for(j=0;j<=2;j++) A[i][j]=C[i][j];
}
void putere(long long sol[3][3],int k)
{
    while(k)
    {
        if(k%2==1)
        {
            inmultire(sol,c);
            k--;
        }
        else
        {
            inmultire(c,c);
            k=k/2;
        }
    }
}
int main()
{
    int x,y,z,A,b,C,t;
    f>>t;
    for(int l=1;l<=t;l++)
    {
        f>>x>>y>>z>>A>>b>>C>>n;
        a[0][0]=x;
        a[0][1]=y;
        a[0][2]=z;
        a[1][0]=0;
        a[1][1]=0;
        a[1][2]=0;
        a[2][0]=0;
        a[2][1]=0;
        a[2][2]=0;

        c[0][0]=0;
        c[0][1]=0;
        c[0][2]=C;
        c[1][0]=1;
        c[1][1]=0;
        c[1][2]=b;
        c[2][0]=0;
        c[2][1]=1;
        c[2][2]=A;

        sol[0][0]=1;
        sol[0][1]=0;
        sol[0][2]=0;
        sol[1][0]=0;
        sol[1][1]=1;
        sol[1][2]=0;
        sol[2][0]=0;
        sol[2][1]=0;
        sol[2][2]=1;

    putere(sol,n);
    inmultire(a,sol);
    g<<a[0][0]<<'\n';

    }

    return 0;
}