字典树模板 HDU

来源：互联网发布：太极熊猫 mac 编辑：程序博客网时间：2024/05/16 14:01

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

#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;}

阅读全文

0 0