hiho 1014

输入

输入的第一行为一个正整数n，表示词典的大小，其后n行，每一行一个单词（不保证是英文单词，也有可能是火星文单词哦），单词由不超过10个的小写英文字母组成，可能存在相同的单词，此时应将其视作不同的单词。接下来的一行为一个正整数m，表示小Hi询问的次数，其后m行，每一行一个字符串，该字符串由不超过10个的小写英文字母组成，表示小Hi的一个询问。
在20%的数据中n, m<=10，词典的字母表大小<=2.
在60%的数据中n, m<=1000，词典的字母表大小<=5.
在100%的数据中n, m<=100000，词典的字母表大小<=26.
本题按通过的数据量排名哦～

输出

对于小Hi的每一个询问，输出一个整数Ans,表示词典中以小Hi给出的字符串为前缀的单词的个数。

样例输入

5
babaab
babbbaaaa
abba
aaaaabaa
babaababb
5
babb
baabaaa
bab
bb
bbabbaab

样例输出

1
0
3
0
0

#include <iostream>
#include <cstring>
using namespace std;
const int MAXN=105;
struct Trie{
char a;
int cnt;
Trie *next[26];
Trie()
{
cnt=0;
memset(next,NULL,sizeof(next));
}
};
Trie root;
int ans;

void Insert(char *str)
{
int len=strlen(str);
Trie *head=&root;
for(int i=0;i<len;i++)
{
if(head->next[str[i]-'a']==NULL)
{
Trie *node=new Trie;
node->a=str[i];
head->next[str[i]-'a']=node;
}
head=head->next[str[i]-'a'];
head->cnt++;
}

}

void find(char *str)
{
int len=strlen(str);
Trie *head=&root;
ans=0;
for(int i=0;i<len;i++)
{
if(head->next[str[i]-'a']==NULL)
return;
else
head=head->next[str[i]-'a'];
}
ans=head->cnt;
}
int main()
{
char str[MAXN];
int n,m;
cin>>n;
for(int i=0;i<n;i++)
{
cin>>str;
Insert(str);
}
cin>>m;
for(int j=0;j<m;j++)
{
cin>>str;
ans=0;
find(str);
cout<<ans<<endl;
}

return 0;
}