Cod sursa(job #1686723)

Utilizator andrew_assassin789Andrei Manea andrew_assassin789 Data 12 aprilie 2016 13:20:20
Problema Trie Scor 90
Compilator cpp Status done
Runda Arhiva educationala Marime 2.19 kb
#include <fstream>
#include <vector>
#include <cstring>
#include <queue>
#define nmax 100100
using namespace std;
struct ntrie
{
    unsigned int nod;
    char c;
};
unsigned int nou,op,rez;
vector <ntrie> T[nmax];
char s[35];
unsigned int frec[nmax];
unsigned int ex[nmax];
void make0()
{
    unsigned int i,n,x=0,j;bool ok;
    n=strlen(s);ntrie y;
    for (i=0;i<n;i++)
    {
        for (j=0,ok=1;j<T[x].size()&&ok;j++)
        {
            if (T[x][j].c==s[i])
                ok=0,x=T[x][j].nod;
        }
        if (ok)
        {
            nou++;
            y.c=s[i];y.nod=nou;
            T[x].push_back(y);
            x=nou;
        }
        ex[x]++;
    }
    frec[x]++;
}
void make2()
{
    unsigned int i,n,x=0,j;bool ok=0;
    n=strlen(s);rez=0;
    for (i=0;i<n&&(!ok);i++)
    {
        for (j=0,ok=1;j<T[x].size()&&ok;j++)
        {
            if (T[x][j].c==s[i])
                ok=0,x=T[x][j].nod;
        }
        if (op==1) ex[x]--;
    }
    if (op==1) frec[x]--;
    else rez=frec[x]*(ok==false);
}
queue <unsigned int> q;
void make3()
{
    unsigned int i,n,x=0,j,nr=0;bool ok=0;
    n=strlen(s);rez=0;
    for (i=0;i<n&&(!ok);i++)
    {
        for (j=0,ok=1;j<T[x].size()&&ok;j++)
        {
            if (T[x][j].c==s[i])
                ok=0,x=T[x][j].nod;
        }
        nr++;
        if (ex[x]&&(!ok)) rez=max(rez,nr);
    }
}
int main()
{
    ifstream f("trie.in");
    ofstream g("trie.out");
    ex[0]=1;unsigned int n,i=0,j=0;
    while (!f.eof())
    {
        f>>op>>s;j++;
        n=strlen(s);
        if (n)
        {
            if (op==0)
            {
                make0();
            }
            else
                if ( op==2 || op==1 )
                {
                    make2();
                    if (op==2)
                    {
                        g<<rez<<'\n';
                        i++;
                    }
                }
                else
                {
                    make3();
                    g<<rez<<'\n';
                    i++;
                }
        }
        memset(s,0,n+1);
    }
    f.close();
    g.close();
    return 0;
}