hdu 1251

最新推荐文章于 2021-11-09 00:20:03 发布

JustSteps

最新推荐文章于 2021-11-09 00:20:03 发布

阅读量369

点赞数

CC 4.0 BY-SA版权

分类专栏： acm字典树

本文链接：https://blog.youkuaiyun.com/JustSteps/article/details/8529944

acm字典树专栏收录该内容

6 篇文章

订阅专栏

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
struct node
{
    node* next[28];
    int count;
    node():count(0){memset(next,0,sizeof(next));}
};
node *root=new node;
void Insert(char *str)
{
    int i;
    node *t=root;
    int l=strlen(str);
    for(i=0;i<l;i++)
    {
        if(t->next[str[i]-'a']==0)
        {
            node *p=new node;
            t->next[str[i]-'a']=p;
            p->count++;
            t=t->next[str[i]-'a'];
        }
        else
        {
            t->next[str[i]-'a']->count++;
            t=t->next[str[i]-'a'];
        }
    }
}
void Find(char *str)
{
    int i,l=strlen(str);
    node *t=root;
    for(i=0;i<l;i++)
    {
        t=t->next[str[i]-'a'];
        if(t==0)
        {
            printf("0\n");
            return ;
        }
    }
    printf("%d\n",t->count);
}

int main()
{

    char str[15];
    while(gets(str))
    {
        if(str[0]=='\0')
            break;
        else
            Insert(str);
    }
    while(gets(str))
    {
        if(str[0]=='\0')
            break;
        else
            Find(str);
    }
    return 0;
}