Cod sursa(job #2252639)

Utilizator Andrei_CotorAndrei Cotor Andrei_Cotor Data 2 octombrie 2018 21:20:34
Problema Pedefe Scor 100
Compilator cpp Status done
Runda Lista lui wefgef Marime 1.69 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[505],rez,i,j,k,Val[505];

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

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

int qry(int poz)
{
    int rez=0;
    while(poz>=1)
    {
        adu(rez,F[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;
        Val[0]=Dp[cr][0][0];
        for(j=1; j<=n; j++)
        {
            for(k=1; k<=501; k++)
                F[k]=0;
            for(k=1; k<=m; k++)
            {
                upd(S2[k-1]+1,Val[k-1]);
                Dp[1-cr][j][k]=0;
                if(S1[j]==S2[k])
                {
                    if(i<k && S1[j]==S3[i+1])
                        adu(Dp[1-cr][j][k],qry(S1[j]+1));
                    else
                        adu(Dp[cr][j][k],qry(S1[j]+1));
                }
                adu(Val[k],Dp[cr][j-1][k]);
            }

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