hihoCoder 1014 Trie树(字典树)

#include<cstdio>
struct trie //利用结构体来封装字典树的节点。
{
trie* next[27];
int num;
trie()
//构造函数。
{
for(int i=0; i<26; i++)
next[i]=NULL;
num=0;
}
} root;
void insert(char* s) //将字符串s所表示的单词插入到字典树中。
{
trie *p=&root;
for(int i=0; s[i]!='\0'; i++)
{
if(p->next[s[i]-'a']==NULL)
p->next[s[i]-'a']=new trie;
p=p->next[s[i]-'a'];
p->num++;
}
}
int find(char *s) //返回值是以s为前缀的单词的数量。
{
trie *p=&root;
for(int i=0; s[i]!='\0'; i++)
if(p->next[s[i]-'a']==NULL)
return 0;
else
p=p->next[s[i]-'a'];
return p->num;
}
int main()
{
char str[100010];
int n;
scanf("%d",&n);
getchar();
while(n--)
{
gets(str);
insert(str);
}
int m;
scanf("%d",&m);
while(m--)
{
scanf("%s",str);
printf("%d\n",find(str));
}
}