Cod sursa(job #2261046)

Utilizator alex2209alexPavel Alexandru alex2209alex Data 15 octombrie 2018 21:15:53
Problema Trie Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.57 kb
#include <fstream>
#define ch (*s-'a')
#include <cstring>
using namespace std;
ifstream f("trie.in");
ofstream g("trie.out");
struct Trie
{
    int pr,rasp;
    Trie *fii[26];
};
Trie *t=new Trie();
int i,ma,c,rasp,nr,n;
char s[26];
void adauga(Trie *p,char *s)
{
    p->pr++;
    if(*s=='1')
    {
        p->rasp++;
        return;
    }
    else
    {
        if(p->fii[ch]==NULL)
        {
            p->fii[ch]=new Trie();
        }
        adauga(p->fii[ch],s+1);
    }
}
void elimina(Trie *p,char *s)
{
    p->pr--;
    if(*s=='1')
    {
        p->rasp--;
    }
    else
    {
        elimina(p->fii[ch],s+1);
    }
}
void calculeaza(Trie *p,char *s)
{
    if(*s=='1')
    {
        g<<p->rasp<<'\n';
        return;
    }
    if(p->fii[ch]==NULL)
    {
        g<<0<<'\n';
        return;
    }
    calculeaza(p->fii[ch],s+1);
}
void cauta(Trie *p,char *s)
{
    if(p->fii[ch]!=NULL && p->fii[ch]->pr>0)
    {
        ma++;
        cauta(p->fii[ch],s+1);
    }
    else
    {
        g<<ma<<'\n';
    }
}
int main()
{
    while(f>>c)
    {
        f.get();
        f.getline(s,30);
        n=strlen(s);
        s[n]='1';
        ma=0;
        if(c==0)
        {
            adauga(t,s);
        }
        else if(c==1)
        {
            elimina(t,s);
        }
        else if(c==2)
        {
            calculeaza(t,s);
        }
        else
        {
            ma=0;
            cauta(t,s);
        }
        for(i=0;i<=n;i++)
        {
            s[i]=0;
        }
    }
    return 0;
}