Cod sursa(job #2297471)

Utilizator iulius510iulius alexandru iulius510 Data 5 decembrie 2018 21:25:24
Problema Trie Scor 90
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 Trie
{ public:
 int nr_cuvinte,sons;
 Trie *fii[26];
  
  Trie()
    {nr_cuvinte=sons=0;
	 for(int i=0;i<=25;i++)
	   fii[i]=0;
    }
	
}*rad=new Trie();
void ad_cuv(char *w)
{  Trie *q=rad;
   for(int i=0;i<strlen(w);i++)
	   {
	   	 if(q->fii[w[i]-'a']==0)
	     {q->fii[w[i]-'a']=new Trie();
          ++q->sons;
         }
         q=q->fii[w[i]-'a'];
	   }
	
	++q->nr_cuvinte;
	
}
bool del_cuv(int poz,Trie *q,char *w)
{ 
	if(poz==strlen(w))
	{--q->nr_cuvinte;
	  if(q->nr_cuvinte==0&&q->sons==0)
	  {   delete q;
	      return 1;
	  }
	
	   return 0;
	 }
	
	if(del_cuv(poz+1,q->fii[w[poz]-'a'],w))
	 {
	 	--q->sons;
	 	q->fii[w[poz]-'a']=0;
	 }
	 if(q!=rad&&(q->sons==0)&&(q->nr_cuvinte==0))
	 {
	 	delete q;
	 	return 1;
	 }
  return 0;
}
void print_nrcuv (char *w)
{   
	Trie *q=rad; int i=0;
   for(i=0;i<strlen(w)&&q->fii[w[i]-'a']!=0;i++)
   	q=q->fii[w[i]-'a'];
   g<<(q->nr_cuvinte)*(i==strlen(w))<<'\n';
}
	
void prefix_maxim(char *w)
{   
    int i=0;
	Trie *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(0,rad,cuv);
			break;
			case 2:
			print_nrcuv(cuv);
			break;
            case 3:
            prefix_maxim(cuv);
            break;
        }
	}
	f.close();
	g.close();
	return 0;
}