Cod sursa(job #2297132)

Utilizator iulius510iulius alexandru iulius510 Data 5 decembrie 2018 14:33:19
Problema Trie Scor 5
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.55 kb
#include<fstream>
	
#include<string.h>
	
using namespace std;
	
ifstream f("trie.in");
	
ofstream g("trie.out");
	
int x;
	
char cuv[20];
	
class nod
	
{  public:
	
	int nr_cuvinte;
	
	nod *fii[26];
	
}*rad=new nod;
	
void ad_cuv(char *w)
	
{
	
	nod *q=rad;
	
	for(int i=0;i<strlen(w);i++)
	
	{  
	
       if(q->fii[w[i]-'a']==0)
	
       	{ q->fii[w[i]-'a']=new nod;
	
          (q->fii[w[i]-'a'])->nr_cuvinte=0;
	
          for(int j=0;j<=25;j++)
	
          	(q->fii[w[i]-'a'])->fii[j]=0;
	
        }
	
       	q=q->fii[w[i]-'a'];
	
	
	}
	
	++q->nr_cuvinte;
	
}
int empty(nod *q)
{
	for(int i=0;i<=25;i++)
	 if(q->fii[i]!=0)
	 	return 0;
	 return 1;
}
	
void del_cuv(char *w,nod *r,nod *q,int pas1,int pas2)
	
{
	
	if(pas2==strlen(w))
	
		--q->nr_cuvinte;
	
	if(pas2<strlen(w))
	
		del_cuv(w,q,q->fii[w[pas2]-'a'],pas2,pas2+1);
	
	
	if(empty(q)&&q!=rad)
	
	{   
	
		r->fii[w[pas1]-'a']=0;
	
		delete q;
	
	}
	
}
	
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);
	
			break;
	
			case 1:
	
			del_cuv(cuv,rad,rad,0,0);
	
			break;
	
			case 2:
	
			print_nrcuv(cuv);
	
			break;
	
			case 3:
	
			prefix_maxim(cuv);
	
			break;
	
		}
	
	}
	
	f.close();
	
	g.close();
	
  return 0;
	
}