Pagini recente » Cod sursa (job #1139810) | Cod sursa (job #2748389) | Cod sursa (job #2296644)
#include<fstream>
#include<string.h>
using namespace std;
ifstream f("trie.in");
ofstream g("trie.out");
int x;
char cuv[20];
struct nod
{ int nr_cuvinte,ex_cuv;
nod *fii[29];
nod()
{
nr_cuvinte=ex_cuv=0;
for(int i=1;i<=26;i++)fii[i]=NULL;
}
}*rad=new nod();
void ad_cuv(char *w,nod *q,int poz)
{
++q->ex_cuv;
if(poz==strlen(w))
q->nr_cuvinte=q->nr_cuvinte+1;
if(poz<strlen(w))
{
if(q->fii[w[poz]-'a']==NULL)
q->fii[w[poz]-'a']=new nod();
ad_cuv(w,q->fii[w[poz]-'a'],poz+1);
}
}
bool del_cuv(int poz,nod *q,char *w)
{
if(poz==strlen(w))
{
q->nr_cuvinte=q->nr_cuvinte-1;
q->ex_cuv=q->ex_cuv-1;
if(q->ex_cuv==0)
{
delete q;
return 1;
}
return 0;
}
if(del_cuv(poz+1,q->fii[w[poz]-'a'],w))
q->fii[w[poz]-'a']=NULL;
if(q!=rad)
{
--q->ex_cuv;
if(q->ex_cuv==0)
{
delete q;
return 1;
}
}
return 0;
}
void print_nrcuv (char *w)
{
nod *aux=rad; int i;
for(i=0;i<strlen(w)&&aux->fii[w[i]-'a']!=NULL;aux=aux->fii[w[i]-'a'],i++);
g<<(aux->nr_cuvinte)*(i==strlen(w))<<'\n';
}
void prefix_maxim(char *w)
{
int i=0;
nod *aux=rad;
for(i=0;i<strlen(w)&&aux->fii[w[i]-'a']!=NULL;aux=aux->fii[w[i]-'a'],i++);
g<<i<<'\n';
}
int main()
{
while(f>>x>>cuv)
{
switch(x)
{
case 0:
ad_cuv(cuv,rad,0);
break;
case 1:
del_cuv(0,rad,cuv);
break;
case 2:
print_nrcuv(cuv);
break;
case 3:
prefix_maxim(cuv);
break;
}
}
f.close();
g.close();
return 0;
}