[复习][poj2001]字典树（trie树）Shortest Prefixes

题目背景

POJ2001

题目描述

给出 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树模板

代码：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<ctime>
#include<cmath>
#include<algorithm>
#include<cctype>
using namespace std;

struct node{
int num;
int son[26];
}trie[200020];
char s[25],c[1010][25];
int cnt,len,tot;

void build_trie()
{
int position=0;
for(int i=0;i<len;++i)
{
if(!trie[position].son[s[i]-'a'])
trie[position].son[s[i]-'a']=++tot;

position=trie[position].son[s[i]-'a'];
trie[position].num++;
}
}

void find_trie()
{
int position=0;
for(int i=0;i<len;++i)
{
if(trie[trie[position].son[s[i]-'a']].num!=1&&i!=len-1)
{
printf("%c",s[i]);
position=trie[position].son[s[i]-'a'];
}
else
{
printf("%c\n",s[i]);
break;
}
}
}

int main()
{
freopen("sp.in","r",stdin);
freopen("sp.out","w",stdout);

while(scanf("%s",s)!=EOF)
{
cnt++;
len=strlen(s);
for(int i=0;i<len;++i)
c[cnt][i]=s[i];
build_trie();
memset(s,0,sizeof(s));
}

for(int i=1;i<=cnt;++i)
{
int j=0;
memset(s,0,sizeof(s));
while(c[i][j]!=0)
{
printf("%c",c[i][j]);
s[j]=c[i][j];
j++;
}

printf(" ");
len=strlen(s);
find_trie();
}

return 0;
}

本题结。