[复习][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;
}

本题结。