trie树模板

指针实现：

class trie  
{  
public:  
    trie* next[26];  
    int num; //记录前缀的个数  
    bool value;//标记这里是不是一个单词  
    trie()    
    {  
        for(int i=0;i<26;i++)  
            next[i]=0;      
        value=0;  
        num=0;    
    }    
}root;  
void insert(char* s)   
{  
    trie *p=&root;  
    int k=0;  
    while(s[k]!='\0')    
    {  
        if(!p->next[s[k]-'a'])  
            p->next[s[k]-'a']=new trie;  
        p=p->next[s[k]-'a'];  
        p->num++;  //每访问节点就++1次，多好啊！  
        k++;    
  
    }  
    p->value=1; //在这里可根据具体题意来加东西  
}  
int find(char *s)   
{  
    trie* p=&root;  
    int k=0;  
    while(s[k]!='\0'&&p->next[s[k]-'a'])   
    {  
        p=p->next[s[k]-'a'];  
        k++;   
    }  
    if(s[k]=='\0')  
        return p->num;  
    return 0;   
}  

静态数组实现，直接开一个很大的数组：

struct trie
{
    int ch[maxnode][sigma_size];
    int val[maxnode];//附加信息
    int sz;
    trie(){sz=1;memset(ch[0],0,sizeof(ch[0]));}
    int idx(char c){return c-'a';}
    void insert(char *s,int v)
    {
        int len=strlen(s),u=0;
        for(int i=0;i<len;i++)
        {
            int c=idx(s[i]);
            if(!ch[u][c])
            {
                memset(ch[sz],0,sizeof(ch[sz]));
                val[sz]=0;
                ch[u][c]=sz++;
            }
            u=ch[u][c];
        }
        val[u]=v;
    }
    bool find(char *s)
    {
        int u=0,len=strlen(s),c;
        for(int i=0;i<len;i++)
        {
            c=idx(s[i]);
            if(!ch[u][c])return false;
            u=ch[u][c];
        }
        if(!val[u])return false;
        return true;
    }
};