Cod sursa(job #2294209)

Utilizator iulius510iulius alexandru iulius510 Data 2 decembrie 2018 01:23:36
Problema Trie Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.47 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,pref_cuv;
	nod *fii[26],*tata;
}*rad=new nod;
void ad_cuv(char *w)
{   
	nod *aux=rad;
	for(int i=0;i<strlen(w);i++)
		if(aux->fii[w[i]-'a']==NULL)
		{   nod *q=new nod;
			q->tata=aux;
			aux->fii[w[i]-'a']=q;
			aux=aux->fii[w[i]-'a'];
			++aux->pref_cuv;
		}
        else
         {
         	aux=aux->fii[w[i]-'a'];
		     ++aux->pref_cuv;
		 }
	
	++aux->nr_cuvinte;
}
void delete_cuv(char *w)
{
	nod *aux=rad;
	for(int i=0;i<strlen(w);i++)
	{   nod *q=aux;
		aux=aux->fii[w[i]-'a'];
	    if((q->pref_cuv==0)&&q!=rad)
	       {q->tata->fii[w[i-1]-'a']==NULL; 
	         delete(q);
	        }
		--aux->pref_cuv;
	}
	--aux->nr_cuvinte;
	if(aux->pref_cuv==0&&aux!=rad)
		{aux->tata->fii[w[strlen(w)-1]-'a']=NULL; 
		  delete(aux);
		}
}
void print_nrcuv (char *w)
{
	nod *aux=rad; int i;
	for(i=0;i<strlen(w)&&aux->fii[w[i]-'a']!=NULL;i++)
	{
		aux=aux->fii[w[i]-'a'];
	}
	g<<aux->nr_cuvinte*(i==strlen(w))<<'\n';
}
void prefix_maxim(char *w)
{
	int nr=0;
	nod *aux=rad;
	for(int i=0;i<strlen(w)&&aux->fii[w[i]-'a']!=NULL;aux=aux->fii[w[i]-'a'],i++)
	      ++nr;
    g<<nr<<'\n';
  
}
int main()
{
	while(f>>x>>cuv)
	{   
		switch(x)
		{
			case 0:
			ad_cuv(cuv);
			break;
			case 1:
			delete_cuv(cuv);
			break;
			case 2:
			print_nrcuv(cuv);
			break;
			case 3:
			prefix_maxim(cuv);
			break;
		}
	}
	f.close();
	g.close();
  return 0;
}