Cod sursa(job #2296775)

Utilizator iulius510iulius alexandru iulius510 Data 5 decembrie 2018 00:21:51
Problema Trie Scor 90
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.39 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,ex_cuv;
	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;
          (q->fii[w[i]-'a'])->ex_cuv=0;
          for(int j=0;j<=25;j++)
          	(q->fii[w[i]-'a'])->fii[j]=0;
        }
       	q=q->fii[w[i]-'a'];
       	++q->ex_cuv;
	}
	++q->nr_cuvinte;
}
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);
	--q->ex_cuv;
	if(q->ex_cuv==0&&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;
}