zoj 1109 Language of FatMouse

字典树实现，也可以map实现

#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <ctype.h>
#include <math.h>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;
#define ll long long
#define ls rt<<1
#define rs ls1
#define lson l,mid,ls
#define rson mid+1,r,rs
#define middle (l+r)>>1
#define eps (1e-9)
#define clr_all(x,c) memset(x,c,sizeof(x))
#define clr(x,c,n) memset(x,c,sizeof(x[0])*(n+1))
#define MOD 1000000007
#define inf 100000007
#define pi acos(-1.0)
#define M 100000+5
#define N 26
typedef struct Trie{
Trie *son[N];
int v;
int w;
};
Trie *root;
char w[M][12]; 
Trie *createNode(){
Trie *p;int i;
p=(Trie *)malloc(sizeof(Trie));
p->v=1;p->w=-1;
for(i=0;i<N;i++)p->son[i]=NULL;
return p;
}
void release(Trie *p){
int i;
for(i=0;i<N;i++)
if(p->son[i]!=NULL)release(p->son[i]);
free(p);
}
void insert(char *s0,int x){
Trie *p=root,*q;int i,len=strlen(s0);
for(i=0;i<len;i++){
int id=s0[i]-'a';
if(p->son[id]==NULL){
q=createNode();
p->son[id]=q;
}else  
p->son[id]->v++;
p=p->son[id];
}
p->w=x;
}
void search(char *st){
Trie *p=root;int i,len=strlen(st);
for(i=0;i<len;i++){
int id=st[i]-'a';
if(p!=NULL)
p=p->son[id];
else if(p==NULL&&i<len-1){
puts("eh");
return;
}
}
puts(w[p->w]);
}
void getWord(char *s1,char *s2,char *s0){
int i,j,len=strlen(s1);
i=0,j=0;
while(s1[i]!=' ')s2[j++]=s1[i++];
s2[j]='\0';
i++;j=0;
while(s1[i]!='\0')s0[j++]=s1[i++];
s0[j]='\0';
}
int main(){
    int i,j;char s0[25],s1[12],s2[12];
    root=createNode();
    gets(s0);i=0;
    while(strcmp(s0,"")!=0){
getWord(s0,s1,s2);
strcpy(w[i],s1);
insert(s2,i);
gets(s0);
i++;
}
while(gets(s0)!=NULL){
search(s0);
}
release(root);
    return 0;
}