某Trie树的简单例题

初学Trie树！解决字符串前缀问题的法宝！

T1：poj2001:Shortest Prefixes

题意：给一堆字符串，问每个字符串最短的不与其他字符串前缀重复的前缀（好奇怪的表述QAQ

貌似“字符串前缀”之类的先考虑Trie树就好了。。。

思路：建Trie树，每路过一个点该点cnt++；最后查询时发现cnt==1的点则为目标，如果找到最后还不行，就直接输出。

代码：

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<iostream>
#define cls(x) memset(x,0,sizeof x)
const int maxn = 1010;
using namespace std;
struct trie
{
int next[27];
int cnt;
}node[maxn<<5];
char c[maxn][25];
int tot;
void add(char* str)
{
int rt=0;
for(int i=0;str[i]!='\0';i++)
{
int x=str[i]-'a';
if(node[rt].next[x])
rt=node[rt].next[x];
else
{
node[rt].next[x]=++tot;
rt=tot;
}
node[rt].cnt++;
}
}
void query(char *str)
{
int rt=0;
char b[100];cls(b);
for(int i=0;str[i]!='\0';i++)
{
int x=str[i]-'a';
if(node[rt].cnt==1)
{
printf("%s\n",b);
return ;
}
rt=node[rt].next[x];
b[i]=str[i];   //Trie树节点是空的，边上是字符，这句话要放在最后而不是前面
}
printf("%s\n",str);
return ;
}
int main()
{
int amt=0;
while(~scanf("%s",c[++amt])) add(c[amt]);
amt--;  // 不要问这是什么。。。EOF时amt也++了
for(int i=1;i<=amt;i++)
{
printf("%s ",c[i]);
query(c[i]);
}
return 0;
}