字典树
Problem Description
遇到单词不认识怎么办? 查字典啊,已知字典中有n个单词,假设单词都是由小写字母组成。现有m个不认识的单词,询问这m个单词是否出现在字典中。
Input
含有多组测试用例。
第一行输入n,m (n>=0&&n<=100000&&m>=0&&m<=100000)分别是字典中存在的n个单词和要查询的m个单词.紧跟着n行,代表字典中存在的单词。
然后m行,要查询的m个单词
n=0&&m=0 程序结束
数据保证所有的单词都是有小写字母组成,并且长度不超过10
Output
若存在则输出Yes,不存在输出No .
Sample Input
3 2
aab
aa
ad
ac
ad
0 0
Sample Output
No
Yes
这个代码是看的一个大佬的,它用的结构体,但是没有用指针。
我觉的巧妙之处在于top和root。
top使建立新节点时对数组的运用更为紧凑,同时使这个值为下一个节点的下标,这样查找的时候就可以直接根据下标查找了,有点链式向前星的感觉。
而root的值,在主函数中一直为根节点的下表也就是0,但是它在建立和查找函数中是不断改变的。
它还有一个非常巧妙的地方,就是如何确定的这个单词是否存在。当这个单词最后一个节点的mark值为真时,则这个单词存在。否则可能不能达到这个尾字母,或者这个单词的尾字母为其他单词的路径。
我不会画图,有图的话可能会好理解些。
给你看看截图吧…
#include <bits/stdc++.h>
using namespace std;
struct node
{
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)
{
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,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);
if(search(root,b))
printf("Yes\n");
else
printf("No\n");
}
}
return 0;
}