字典树 SDUT OJ 2828

Time Limit: 1000MS Memory limit: 65536K

题目描述

遇到单词不认识怎么办? 查字典啊，已知字典中有n个单词，假设单词都是由小写字母组成。现有m个不认识的单词，询问这m个单词是否出现在字典中。

输入

含有多组测试用例。

第一行输入n，m (n>=0&&n<=100000&&m>=0&&m<=100000)分别是字典中存在的n个单词和要查询的m个单词.

紧跟着n行，代表字典中存在的单词。

然后m行，要查询的m个单词

n=0&&m=0 程序结束

数据保证所有的单词都是有小写字母组成，并且长度不超过10

输出

若存在则输出Yes，不存在输出No .

示例输入

3 2
aab
aa
ad
ac
ad
0 0

示例输出

No
Yes

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
struct ooo
{
    int next[26];   
    bool mark;  //标记
}dd[505555];
int top,n,m;
char a[20],b[20];
int creat()  //分配新的节点
{
    memset(dd[top].next,-1,sizeof(dd[top].next));  //表示指向空
    dd[top].mark=false;//表示无对应字符串
    return top++;
}
int xiab(char c)
{
    return c-'a'; //对应下标
}
void insert(int root,char *s)   //将S插入到字典树中
{ 
    int i,len=strlen(s);
    for(i=0;i<len;i++)
    {
        if(dd[root].next[xiab(s[i])]==-1)
        dd[root].next[xiab(s[i])]=creat();
        root=dd[root].next[xiab(s[i])];
    }
    dd[root].mark=true;
}

bool search(int root,char *s)  //询问是否出现在字典树中
{
    for(int i=0;s[i]!='\0';i++)
    {
        if(dd[root].next[xiab(s[i])]==-1)
        return false;
        root=dd[root].next[xiab(s[i])];
    }
    return dd[root].mark;
}
int main()
{
    int i,j,root;

    while(scanf("%d %d",&n,&m)&&(n||m))
    {
        top=0;
        root=creat();
        for(i=0;i<n;i++)
        {
        scanf("%s",a);
        insert(root,a);
        }
        for(i=0;i<m;i++)
        {
           scanf("%s",b);
           printf("%s\n",search(root,b)?"Yes":"No");
        }
    }
    return 0;
}