Pagini recente » Cod sursa (job #663056) | Cod sursa (job #1021064) | Cod sursa (job #2528066) | Cod sursa (job #1934557) | Cod sursa (job #2398170)
#include <bits/stdc++.h>
#define ch (*s-'a')
using namespace std;
struct Trie
{
int nrfii;
int cnt;
Trie *fii[26];
Trie()
{
cnt=0;
nrfii=0;
memset(fii,0,sizeof(fii[0]));
}
};
Trie *T = new Trie;
void ins(Trie *nod,char *s)
{
if(*s=='\n')
{
nod->cnt++;
return ;
}
if(nod->fii[ch]==0)
{
nod->fii[ch]= new Trie;
nod->nrfii++;
}
ins(nod->fii[ch],s+1);
}
int del(Trie *nod,char *s)
{
if(*s=='\n')nod->cnt--;
else{
if(del(nod->fii[ch],s+1))
{
nod->fii[ch]=0;
nod->nrfii--;
}
}
if(nod->nrfii==0 && nod->cnt==0 && nod!=T)
{
delete nod;
return 1;
}
return 0;
}
int apar(Trie *nod,char *s)
{
if(*s=='\n')return nod->cnt;
if(nod->fii[ch])return apar(nod->fii[ch],s+1);
return 0;
}
int pre(Trie *nod,char *s,int k)
{
if(*s=='\n' || nod->fii[ch]==0)return k;
return pre(nod->fii[ch],s+1,k+1);
}
int main()
{
freopen("trie.in","r",stdin);
freopen("trie.out","w",stdout);
char linie[32];
fgets(linie,32,stdin);
while(!feof(stdin))
{
strcat(linie,"\n");
switch(linie[0])
{
case '0':ins(T,linie+2);break;
case '1':del(T,linie+2);break;
case '2':printf("%d\n",apar(T,linie+2));;break;
case '3':printf("%d\n",pre(T,linie+2,0));break;
}
fgets(linie,32,stdin);
}
return 0;
}