字典树 hdu 1251

来源：互联网发布：python iso8601 编辑：程序博客网时间：2024/05/16 04:32
#include<stdio.h>#include<algorithm>#include<stdlib.h>#include<malloc.h>#include<string.h>#include<math.h>using namespace std;struct trie{    trie * next[26];    int bat=0;};int temp=0;int ani=0;int sum=0;trie * init(){    trie *tmp=new trie;    for(int i=0;i<26;i++)    {        tmp->next[i]=NULL;    }    return tmp;}trie *H=init();void insert(char *s){    trie *tmp=H;    int len=strlen(s);    //printf("%d\n",len);    for(int i=0;i<len;i++)    {        int k=s[i]-'a';        if(tmp->next[k]==NULL)        {            tmp->next[k]=init();        }        tmp=tmp->next[k];        tmp->bat++;    }    //tmp->flag=true;}void out1(char *s){    trie *tmp=H;    int len=strlen(s);    for(int i=0;i<len;i++)    {        int k=s[i]-'a';        if(tmp->next[k]!=NULL)        {            tmp=tmp->next[k];        }        else        {            ani=1;            return ;        }    }    int k=s[len-1]-'a';    if(tmp!=NULL)    {        //printf("%d\n",sum);        sum=tmp->bat;        //printf("%d\n",sum);    }    else        ani=1;    //puts("2");    //tt(tmp);    return ;}void del(trie *tmp){    for(int i=0;i<26;i++)    {        if(tmp->next[i]!=NULL)            del(tmp->next[i]);    }    free(tmp);    return ;}int main(){    char s[10217];    while(gets(s))    {        if(s[0]==0&&temp==0)        {            temp=1;            continue;        }        if(s[0]==0&&temp==1)        {            temp=0;            break;        }        if(temp==0)        {            insert(s);        }        else        {            out1(s);            if(ani==1)            {                puts("0");                sum=0;                ani=0;            }            else            {                if(sum==0)                    puts("1");                else                printf("%d\n",sum);                ani=0;                sum=0;            }        }    }    return 0;}
0 0