POJ 2001 TRIE

<pre name="code" class="delphi">#include"stdio.h"
#include"iostream"
#include"string.h"
#include"stdlib.h"
using namespace std;
const int maxn=20*1000+20;
const int max_sigma=26;
const int max_len=30;
char s[1000+20][max_len];
struct trie{
    int node[maxn][max_sigma];
    int val[maxn];
    int cnt;
    void init(){cnt=1;memset(node[0],0,sizeof(node[0]));memset(val,0,sizeof(val));}   //初始化  节点数为1 根节点 计数数组 
    int idx(char c){return c-'a';}
    void insert(char *s)
    {
        int p=0,k;
        for(int i=0;s[i];i++)
        {
            int k=idx(s[i]);          
            if(!node[p][k])
            {
                memset(node[cnt],0,sizeof(node[cnt])); //新建一个节点
                node[p][k]=cnt++;
            }
            p=node[p][k];                        //移动到子节点
            val[p]++;                           //这种前缀的数量加+1
        }
    }
    int find(char *S)  //反值为 字符串的下标
    {
        int k;
        int p=0;
        int i;
        int len=strlen(S);
        for(i=0;S[i];i++)
        {
            k=idx(S[i]);
            if(val[node[p][k]]==1) return i;     //当这种前缀只出现一次说明就是最短前缀 
            p=node[p][k];
        }
        return len-1;                      //当没有这种前缀
    }
};
int main()
{
    //freopen("a.txt","r",stdin);
    int n;
    char tmp[max_len];
    int i,j;
    trie a;
    a.init();
    int cnt=0;
    while(scanf("%s",s[cnt])!=EOF)
    {
        a.insert(s[cnt]);cnt++;
    }
    for(i=0;i<cnt;i++)
    {
        printf("%s ",s[i]);
        int pos=a.find(s[i]);
        for(j=0;j<=pos;j++)
            putchar(s[i][j]);
        puts("");
    }
    return 0;
}