字典树模板 HDU - 1251

字典树模板分为动态分配内存的和数组的写法。动态分配内存的在多组数据的情况下，要销毁内存，很麻烦。建议用数组的。

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <cstring>
#include <queue>
using namespace std;
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
struct node{
    int next[27];
    int v;
    void init(){
        v=0;
        memset(next,-1,sizeof(next));
    }
};
struct node L[1000000];
int tot=0;

void add(char a[],int len){
    int now=0;
    for(int i=0;i<len;i++){
        int tmp=a[i]-'a';
        int next=L[now].next[tmp];
        if(next==-1){
            next=++tot;
            L[next].v=-1;
            L[next].init();
            L[now].next[tmp]=next;
        }
        now=next;
        L[now].v++;
    }

}

int query(char a[],int len){
    int now=0;
    for(int i=0;i<len;i++){
        int tmp=a[i]-'a';
        int next=L[now].next[tmp];
        if(next==-1)return 0;
        now=next;
    }
    return L[now].v;
}
int main()
{
    char str[20];
    L[0].init();
    while(gets(str),strcmp(str,""))
    add(str,strlen(str));
    while(gets(str)!=NULL)
    printf("%d\n",query(str, strlen(str)));
    return 0;
}