Cod sursa(job #2252603)

Utilizator Andrei_CotorAndrei Cotor Andrei_Cotor Data 2 octombrie 2018 20:43:49
Problema Pedefe Scor 75
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.91 kb
#include<fstream>
#define MOD 666013
using namespace std;
ifstream fi("pedefe.in");
ofstream fo("pedefe.out");
int n,m,p,S1[505],S2[505],S3[505],cr,Dp[2][505][505],F[2][505],rez,i,j,k,Val[2][505];

void adu(int &x, int y)
{
    int rez=x+y;
    if(rez>=MOD)
        rez-=MOD;
    x=rez;
}

void upd(int cr, int poz, int val)
{
    if(val==0)
        return;
    while(poz<=501)
    {
        adu(F[cr][poz],val);
        poz=poz+(poz&(poz^(poz-1)));
    }
}

int qry(int cr, int poz)
{
    int rez=0;
    while(poz>=1)
    {
        adu(rez,F[cr][poz]);
        poz=poz-(poz&(poz^(poz-1)));
    }
    return rez;
}

int main()
{
    fi>>n>>m>>p;
    for(i=1; i<=n; i++)
        fi>>S1[i];
    for(i=1; i<=m; i++)
        fi>>S2[i];
    for(i=1; i<=p; i++)
        fi>>S3[i];
    Dp[0][0][0]=1;
    cr=1;
    for(i=0; i<=p; i++)
    {
        cr=1-cr;
        for(j=1; j<=n; j++)
        {
            Val[cr][0]=Dp[cr][0][0];
            Val[1-cr][0]=Dp[1-cr][0][0];
            for(k=1; k<=501; k++)
                F[0][k]=F[1][k]=0;
            for(k=1; k<=m; k++)
            {
                upd(cr,S2[k-1]+1,Val[cr][k-1]);
                upd(1-cr,S2[k-1]+1,Val[1-cr][k-1]);
                if(S1[j]==S2[k])
                {
                    if(S1[j]==S3[i])
                        Dp[cr][j][k]=qry(1-cr,S1[j]+1);
                    else
                        Dp[cr][j][k]=qry(cr,S1[j]+1);
                    if(i==p)
                        adu(rez,Dp[cr][j][k]);
                }
                else
                    Dp[cr][j][k]=0;
                adu(Val[cr][k],Dp[cr][j-1][k]);
                adu(Val[1-cr][k],Dp[1-cr][j-1][k]);
            }

        }
        if(i==1)
            Dp[0][0][0]=0;
        for(j=1; j<=m; j++)
            Val[1-cr][j]=Val[cr][j]=0;
    }
    fo<<rez<<"\n";
    fi.close();
    fo.close();
    return 0;
}