Cod sursa(job #1471341)

Utilizator BlaugranasEnal Gemaledin Blaugranas Data 13 august 2015 17:04:20
Problema Trie Scor 25
Compilator c Status done
Runda Arhiva educationala Marime 1.09 kb
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
typedef struct M {
	int c,n;
	struct M *f[26];
}N;
char l[32];
void A(N *s,char *l) {
	if(*l=='\n') {
		s->c++;
        return;
	}
	if(!s->f[*l-'a'])
        s->f[*l-'a']=(N*)malloc(sizeof(N)),s->n++;
	A(s->f[*l-'a'],l+1);
}
int D(N *s,char *l) {
	if(*l=='\n')
        s->c--;
	else if(D(s->f[*l-'a'],l+1))
        s->f[*l-'a']=0,s->n--;
	if(!s->c&&!s->n) {
		free(s);
      	return 1;
	}
	return 0;
}
int C(N *s,char *l) {
	if(*l=='\n')
        return s->c;
	if(s->f[*l-'a'])
        return C(s->f[*l-'a'],l+1);
	return 0;
}
int E(N *s,char *l,int k) { return *l=='\n'||!s->f[*l-'a']?k:E(s->f[*l-'a'],l+1,k+1); }
int main() {
	freopen("trie.in","r",stdin),freopen("trie.out","w",stdout);
	N *s=(N*)malloc(sizeof(N));
	s->c=s->n=0,memset(s->f,0,sizeof(s->f));
	do {
        fgets(l,32,stdin);
		if(l[0]=='0')
        	A(s,l+2);
     	if(l[0]=='1')
            D(s,l+2);
     	if(l[0]=='2')
            printf("%d\n",C(s,l+2));
     	if(l[0]=='3')
            printf("%d\n",E(s,l+2,0));
	}while(!feof(stdin));
}