Cod sursa(job #1887678)

Utilizator delta_wolfAndrei Stoica delta_wolf Data 21 februarie 2017 18:35:20
Problema Trie Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.39 kb
#include <cstdio>
#include<cstring>
using namespace std;
int n,op;
char s[25];
struct tri
{
    int cuv, pref;
    tri *fii[27];
    tri()
    {
        cuv=pref=0;
        memset(fii,0,sizeof(fii));
    }

}*trie;


void add(char s[])
{
    tri *q=trie;
    trie->pref++;
    for(int i=0;i<n;i++)
    {
        if(q->fii[s[i]]==0)
            q->fii[s[i]]=new tri;
        q=q->fii[s[i]];
        q->pref++;
    }
    q->cuv++;
}

int cuv(char s[])
{
    tri *q=trie;
    for(int i=0;i<n;i++)
    {
        if(q->fii[s[i]]==0)return 0;
        q=q->fii[s[i]];
    }
    return q->cuv;
}

int pref(char s[])
{
    tri *q=trie;
    for(int i=0;i<n;i++)
    {
        if(q->fii[s[i]]==0||(q->fii[s[i]]->pref==0))return i;
        q=q->fii[s[i]];
    }
    return n;
}

void del(char s[])
{
    tri *q=trie;
    trie->pref--;
    for(int i=0;i<n;i++)
    {
        q=q->fii[s[i]];
        q->pref--;
    }
    q->cuv--;
}

int main()
{
    freopen("trie.in","r",stdin);
    freopen("trie.out","w",stdout);

    trie= new tri;

    while(scanf("%d %s",&op,&s)!=EOF)
    {
        n=strlen(s);
        for(int i=0;i<n;i++)
            s[i]-='a';
        if(op==0)
            add(s);
        else if(op==1)
            del(s);
        else if(op==2)
            printf("%d\n",cuv(s));
        else printf("%d\n",pref(s));
    }
    return 0;
}