初学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;
}