Pagini recente » Cod sursa (job #835312) | Cod sursa (job #49212) | Cod sursa (job #2890193) | Cod sursa (job #2628811) | Cod sursa (job #1647763)
#include <iostream>
#include <cstdio>
#define CH (*s-'a')
#include <cstring>
using namespace std;
struct trie
{
int nrfii,nrcuv;
trie *fiu[27];
trie()
{
nrfii=0; nrcuv=0;
memset(fiu,0,sizeof(fiu));
}
} ;
trie *t=new trie;
void ins(trie *nod,char *s)
{
if(*s=='\n') { nod->nrcuv++; return; }
if(nod->fiu[CH]==0)
{
nod->fiu[CH]=new trie;
nod->nrfii++;
}
ins(nod->fiu[CH],s+1);
}
int del(trie *nod,char *s)
{
if(*s=='\n') nod->nrcuv--;
else if(del(nod->fiu[CH],s+1))
{
nod->fiu[CH]=0;
nod->nrfii--;
}
if(nod->nrcuv==0 && nod->nrfii==0 && nod!=t) { delete nod; return 1; }
return 0;
}
int pref(trie *nod, char *s,int k)
{
if(*s=='\n' || nod->fiu[CH]==0) return k;
return pref(nod->fiu[CH],s+1,k+1);
}
int apar(trie *nod,char *s)
{
if(*s=='\n') return nod->nrcuv;
if(nod->fiu[CH]) return apar(nod->fiu[CH],s+1);
return 0;
}
int main()
{
freopen("trie.in","r",stdin);
freopen("trie.out","w",stdout);
char line[25];
fgets(line,25,stdin);
while(!feof(stdin))
{
if(line[0]=='0') ins(t,line+2);
if(line[0]=='1') del(t,line+2);
if(line[0]=='2') printf("%d\n",apar(t,line+2));
if(line[0]=='3') printf("%d\n",pref(t,line+2,0));
fgets(line,25,stdin);
}
fclose(stdin);
fclose(stdout);
return 0;
}