POJ 2001 字典树

贴个模板。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <cmath>
#include <cstring>
#include <queue>
#include <set>
#include <vector>
#include <stack>
#include <map>
#include <iomanip>
#define PI acos(-1.0)
#define Max 2005
#define inf 1<<28
#define LL(x) (x<<1)
#define RR(x) (x<<1|1)
#define ll long long
using namespace std;

struct Trie
{
    int count;//记录经过这个分支的个数
    struct Trie *next[26];
    Trie()
    {
        count=0;
        memset(next,0,sizeof(next));
    }
};
Trie *root=NULL;
void build_Trie(char *s)//建树
{
    Trie *p=root;
    Trie *temp=NULL;
    int l=strlen(s);
    for(int i=0; i<l; i++)
    {
        int index=s[i]-'a';
        if(p->next[index]==NULL)
        {
            temp=new Trie;
            p->next[index]=temp;
        }
        p=p->next[index];
        p->count++;
    }
}
void Trie_search(char *s)
{
    Trie *p=root;
    int l=strlen(s);
    for(int i=0; i<l; i++)
    {
        int index=s[i]-'a';
        p=p->next[index];
        //cout<<s[i];
        printf("%c",s[i]);
        if(p->count==1)//如果只有一个支路经过该节点，说明这个前缀是唯一的
            break;
    }
    cout<<endl;
}
int main()
{
    char a[1002][22];
    int i=0;
    root =new Trie;
    while(cin>>a[i])
    {
        build_Trie(a[i]);
        i++;
    }
    for(int k=0; k<i; k++)
    {
        printf("%s ",a[k]);
        Trie_search(a[k]);
    }
    return 0;
}