Cod sursa(job #295551)

Utilizator runnaway90Oprescu Radu Constantin runnaway90 Data 3 aprilie 2009 13:26:55
Problema Trie Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.52 kb
#include<stdio.h>
#include<string.h>

#define CH *s - 'a'

struct trie{
    int nrf, cnt;
    trie *fiu[26];
        trie()
        {
            cnt = nrf = 0;
            memset( fiu, 0, sizeof( fiu ) );
        }
};

trie *T= new trie;

void insert(trie *nod,char *s)
{int w = CH;
    if(*s == '\n')
    {
        nod->cnt++;
        return;
    }
    if(! nod->fiu[w])
    {
        nod->fiu[w]= new trie;
        nod->nrf++;
    }
    insert(nod->fiu[w],s+1);
}

int remove(trie *nod,char *s)
{int w = CH;
    if(*s == '\n')
        nod->cnt --;
    else
    if (remove(nod->fiu[w],s+1))
    {
        nod->fiu[w]=0;
        nod->nrf--;
    }
    if (nod->cnt == 0 && nod->nrf == 0 && nod!=T)
    {
        delete nod;
        return 1;
    }
    return 0;
}

int query(trie *nod,char *s)
{int w = CH;
    if(*s== '\n')
        return nod->cnt;
    if(nod->fiu[w])
        return(query(nod->fiu[w],s+1));
    return 0;
}

int prefix(trie *nod,char *s,int level)
{int w = CH;
    if(*s=='\n' || !nod->fiu[w])
        return level;
    return prefix(nod->fiu[w],s+1,level+1);
}

int main()
{
    freopen("trie.in","r",stdin);
    freopen("trie.out","w",stdout);
    char s[32];
    fgets(s,32,stdin);

    while(!feof(stdin))
    {
        if(s[0]=='0') insert(T,s+2);

        if(s[0]=='1') remove(T,s+2);

        if(s[0]=='2') printf("%d\n", query(T,s+2));

        if(s[0]=='3') printf("%d\n", prefix(T,s+2,0));

        fgets(s,32,stdin);
    }
    return 0;
}