Pagini recente » Cod sursa (job #2165322) | Cod sursa (job #350365)
Cod sursa(job #350365)
#include<stdio.h>
#include<string.h>
int m;
char s[25];
struct nod
{
nod *u[28];
long nc,pf;
};
nod *t;
void init(nod *t)
{
int i;
for (i=0;i<26;i++)
t->u[i]=0;
t->nc=0;
t->pf=0;
}
void insert()
{
int i;
nod *p,*aux;
p=t;
for (i=0;i<m-1;i++)
{
if (p->u[s[i]-'a'])
{
p=p->u[s[i]-'a'];
p->pf++;
}
else
{
aux=new nod;
init(aux);
p->u[s[i]-'a']=aux;
p=aux;
p->pf++;
}
}
if (p->u[s[i]-'a'])
{
p=p->u[s[i]-'a'];
p->nc++;
p->pf++;
}
else
{
aux=new nod;
init(aux);
p->u[s[i]-'a']=aux;
p=aux;
p->nc++;
p->pf++;
}
}
void del()
{
int i;
nod *p;
p=t;
for (i=0;i<m;i++)
{
p=p->u[s[i]-'a'];
p->pf--;
}
p->nc--;
}
void nrap()
{
int i;
nod *p;
p=t;
for (i=0;i<m;i++)
{
if (p->u[s[i]-'a'])
p=p->u[s[i]-'a'];
else
{
printf("0\n");
return;
}
}
printf("%ld\n",p->nc);
}
void prefix()
{
int i;
long l=0;
nod *p;
p=t;
for (i=0;i<m;i++)
{
if (p->u[s[i]-'a'])
{
p=p->u[s[i]-'a'];
if (p->pf)
l++;
else
{
printf("%ld\n",l);
return;
}
}
else
{
printf("%ld\n",l);
return;
}
}
}
int main()
{
freopen("trie.in","r",stdin);
freopen("trie.out","w",stdout);
t=new nod;
init(t);
while (scanf("%s",&s)!=EOF)
{
switch (s[0])
{
case '0':{scanf("%s",s);m=strlen(s);insert();break;}
case '1':{scanf("%s",s);m=strlen(s);del();break;}
case '2':{scanf("%s",s);m=strlen(s);nrap();break;}
case '3':{scanf("%s",s);m=strlen(s);prefix();break;}
}
}
return 0;
}