题目描述:
题目链接: POJ 2001 :http://poj.org/problem?id=2001
给出 n 个单词(1<=n<=1000),求出每个单词的非公共前缀,如果没有,则输出自己。
输入格式:
输入 N 个单词,每行一个,每个单词都是由 1~20 个小写字母构成。
输出格式:
输出 N 行,每行由一个空格的两部分,第一部分是输入的单词,第二部分是该单词在所有单词中的非公共前缀,如果没有,则输出自己。
样例输入:
carbohydrate
cart
carburetor
caramel
caribou
carbonic
cartilage
carbon
carriage
carton
car
carbonate
样例输出:
carbohydrate carboh
cart cart
carburetor carbu
caramel cara
caribou cari
carbonic carboni
cartilage carti
carbon carbon
carriage carr
carton carto
car car
carbonate carbona
题目分析:
trie树模板。每次加入都计数,对于满足非公共前缀的,则前部分的num大于1,后面等于1,只需输出到第一个1的位置。对于不满足的则直到单词结尾都大于1,输出全部即可。
附代码:
#include<iostream>
#include<cstring>
#include<string>
#include<cstdlib>
#include<cstdio>
#include<ctime>
#include<queue>
#include<cmath>
#include<cctype>
#include<set>
#include<map>
#include<iomanip>
#include<algorithm>
using namespace std;
int tot,len,num;
char s[26],s1[1010][26];
bool check;
struct node{
int son[26];
int num;
}trie[20010];
void create()
{
int u=0;
len=strlen(s);
for(int i=0;i<len;i++)
{
if(trie[u].son[s[i]-'a']==0)
trie[u].son[s[i]-'a']=++tot;
u=trie[u].son[s[i]-'a'];
trie[u].num++;
}
}
int find()
{
int u=0;
len=strlen(s);
for(int i=0;i<len;i++)
{
u=trie[u].son[s[i]-'a'];
if(check==false&&trie[u].num==1)
return i;
if(trie[u].num>1)
check=true;
if(check==true&&trie[u].num==1)
return i;
if(i==len-1&&trie[u].num>1)
return i;
}
}
int main()
{
//freopen("lx.in","r",stdin);
while(scanf("%s",s)!=EOF)
{
num++;
strcpy(s1[num],s);
create();
}
for(int i=1;i<=num;i++)
{
check=false;
strcpy(s,s1[i]);
printf("%s ",s);
int id=find();
for(int i=0;i<=id;i++)
printf("%c",s[i]);
printf("\n");
}
return 0;
}