Cod sursa(job #2296644)

Utilizator iulius510iulius alexandru iulius510 Data 4 decembrie 2018 21:06:45
Problema Trie Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.42 kb
#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;
}