字典树(Tire)模板

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
struct node
{
    node *ne[26];
    int cnt;
}*head;
void insert(char *s)
{
    node *p=head,*q;
    for(int i=0;s[i];i++)
    {
        int id=s[i]-'a';
        if(p->ne[id]!=NULL)
        {
            p=p->ne[id];
            p->cnt++;
        }
        else
        {
            q=(node*)malloc(sizeof(node));
            for(int i=0;i<26;i++)
                q->ne[i]=NULL;
            q->cnt=1;
            p->ne[id]=q;
            p=p->ne[id];
        }
    }
}
int Q(char *s)//查询有多少以该串为前缀的串的数量
{
    node *p=head;
    for(int i=0;s[i];i++)
    {
        int id=s[i]-'a';
        if(p->ne[id]!=NULL)
            p=p->ne[id];
        else
            return 0;
    }
    return p->cnt;
}
void clean(node *p)//释放空间
{
    if(p==NULL)    return ;
    for(int i=0;i<26;i++)
    {
        if(p->ne[i]!=NULL)
            clean(p->ne[i]);
    }
    free(p);
    return ;
}
int main()
{
    int n,m;
    char s[11];
    while(scanf("%d",&n)!=EOF)
    {
        head=(node*)malloc(sizeof(node));
        for(int i=0;i<26;i++)
            head->ne[i]=NULL;
        while(n--)
        {
            scanf("%s",s);
            insert(s);
        }
        scanf("%d",&m);
        while(m--)
        {
            scanf("%s",s);
            printf("%d
",Q(s));
        }
        clean(head);
    }
    return 0;
}

 hiho1014

#include<stdio.h>
#include<string.h>
const int N=1e5+11;
int t[N*11][27],cnt;
void insert(char *s){
    int now=0;
    for(int i=0;s[i];i++){
        int id=s[i]-'a';
        if(!t[now][id]){
            t[now][id]=++cnt;
        }
        now=t[now][id];
        t[now][26]++;
    }
}
int Q(char *s){
    int now=0;
    for(int i=0;s[i];i++){
        int id=s[i]-'a';
        if(t[now][id])
            now=t[now][id];
        else
            return 0;
    }
    return t[now][26];
}
int main(){
    int n;
    char s[20];
    while(scanf("%d",&n)!=EOF){
        memset(t,0,sizeof(t));cnt=0;
        while(n--){
            scanf("%s",s);
            insert(s);
        }
        scanf("%d",&n);
        while(n--){
            scanf("%s",s);
            printf("%d
",Q(s));
        }
    }
    return 0;
}