字典树 hdu 1075

#include"stdio.h"
#include"ctype.h"
#include"stdlib.h"
#include"string.h"
#include"algorithm"
using namespace std;
struct Trie
{
struct Trie * next[26];
int flag;
};
char xx[1000005][15];   //数组要开1000000，题目没写那么清楚
Trie * H = NULL;
Trie * init()
{
Trie * l = new Trie;
for(int i = 0;i < 26;i ++)
{
l ->next[i] = NULL;
}
l ->flag = 0;   //初始化为0
return l;
}
void Add(char * s,int x)
{
Trie * l = H;
int len = strlen(s);
for(int i = 0;i < len;i ++)
{
int k = s[i] - 'a';
if(l ->next[k] == NULL)
{
l ->next[k] = init();
}
l = l ->next[k];
}
l ->flag = x;    //存起来的时候标记成数组下标 ，查找方便
};
void Cls(Trie * l)
{
for(int i = 0;i < 26;i ++)
{
if(l ->next[i] != NULL)
{
Cls(l ->next[i]);
}
}
free(l);
}
void Search(char * s)
{
Trie * l = H;
int len = strlen(s);
int pl = 1;
for(int i = 0;i < len;i ++)
{
int k = s[i] - 'a';
if(l ->next[k] == NULL)
{
pl = 0;
break;
}
l = l ->next[k];
}
if(l ->flag && pl)  //查找看有没有这个单词，有的话就输出对应下标的字符串，添加时的下标记录用上了
{
printf("%s",xx[l ->flag]);
}
else  //没找到就输出自己
{
printf("%s",s);
}
}
int main()
{
int a,b,c,d,pl = 0;
char vv[30005];
while(~scanf("%s",vv))
{
if(strcmp(vv,"START") == 0 && !pl)   //先保存到数组 ，下标从一开始，因为一会搜索要用
{
H = init();
pl = 1;
int l = 1;
while(1)
{
scanf("%s",xx[l]);
if(strcmp(xx[l],"END") == 0)
break;
else
{
scanf("%s",vv);
Add(vv,l);
l ++;
}
}
}
else
{
getchar();    //加一个gechar()，因为上面是scanf结尾，要去掉一个回车
pl = 0;
while(1)
{
char mm[20];
gets(vv);
if(strcmp(vv,"END") == 0)
break;
else
{
int l = strlen(vv);
int k = 0;
for(int i = 0;i < l;i ++)
{
if(isalpha(vv[i]))
{
mm[k++] = vv[i];
}
else
{
if(k > 0)   //在mm数组里有值得时候才去搜索，以为mm没有初始化
{
mm[k] = '\0';
Search(mm);
}
k = 0;   //个数归零
printf("%c",vv[i]);   //输出不是字母的东西，比如'.','!','?',' '之类的
}
}
printf("\n");
}
}
Cls(H);   //清不清空都行，以为只有一个例子，但我习惯清空了
}
}
return 0;
}