Cod sursa(job #2053040)

Utilizator TherevengerkingSurani Adrian Therevengerking Data 31 octombrie 2017 13:00:41
Problema Trie Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.5 kb
#include <bits/stdc++.h>

using namespace std;
ifstream fin("trie.in");
ofstream fout("trie.out");
typedef struct nod{
    struct nod *fiu[28]; int terminal, nrf;
} NOD, *TRIE;
TRIE T = NULL;
int v, maxim, best;
char cuv[25], *p1;
void adauga(TRIE &T, char *p)
{
    if(T == 0)
    {
        T = new NOD;
        for(int i = 0; i <= 25; ++i)
            T->fiu[i] = 0;
        T->nrf = 0;
        T->terminal = 0;
        adauga(T, p);
    }
    else
    {
        T->nrf++;
        if(*p == '\0')T->terminal ++;
        else adauga(T->fiu[*p - 'a'], p + 1);
    }
}
void sterge(TRIE &T, char *p)
{
    T -> nrf = T -> nrf - 1;
    if((*p) == '\0')
    {
        T -> terminal = (T -> terminal) - 1;
    }
    else sterge(T->fiu[(*p) - 'a'], p + 1);
}
int nrapr(TRIE &T, char *p)
{
    int x = *p - 'a';
    if(*p == '\0')return T->terminal;
    if(T->fiu[x] == 0)return 0;
    return nrapr(T->fiu[x], p + 1);
}
void cautpr(TRIE &T, char *p)
{
    if(T == NULL)
    {
        fout << best - 1 << '\n';
        return;
    }
    if(T -> nrf >= 1)best++;
    if((*p) == '\0')
    {
        fout << best - 1<< '\n';
    }
    else
        cautpr(T->fiu[(*p) - 'a'], p + 1);
}
int main()
{
    while(fin >> v)
    {
        fin >> cuv;
        if(v == 0)adauga(T, cuv);
        else if(v == 1)sterge(T, cuv);
        else if(v == 2)fout << nrapr(T, cuv) << '\n';
        else
        {
            best  = 0;
            cautpr(T, cuv);
        }
    }
    return 0;
}