输入
输入的第一行为一个正整数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给出的字符串为前缀的单词的个数。
#include<iostream>
#include<vector>
#include<string>
using namespace std;
struct Trie{
char key;
int Count;
Trie* Next[26];
}*T;
void buildTree(string str);
void TrieClear(Trie *p);
void InitTrie(int n);
void TrieClear(Trie *p){
for(int i = 0;i < 26;i++){
p->Next[i] = NULL;
}
}
void InitTrie(int n){
T = (Trie*)malloc(sizeof(Trie));
T -> Count = 0;
TrieClear(T);
for(int i = 0;i< n;i++){
string str; cin >> str;
buildTree(str);
}
}
void buildTree(string str){
Trie* p = T;
for(int i = 0;i < str.size(); i++){
if(p->Next[str[i] - 'a'] == NULL){
p->Next[str[i] - 'a'] = (Trie*)malloc(sizeof(Trie));
p = p->Next[str[i] - 'a'];
p->key = str[i];
TrieClear(p);
p->Count = 1;
}else{
p = p->Next[str[i] - 'a'];
p->key = str[i];
p->Count += 1;
}
}
}
int findCount(string str){
Trie *p = T;
for(int i = 0;i < str.size(); i++){
if(p -> Next[str[i] - 'a'] != NULL){
p = p -> Next[str[i] - 'a'];
}else{
return 0;
}
}
return p->Count;
}
int main()
{
int n;
cin >> n;
InitTrie(n);
cin >> n;
for(int i = 0; i < n;i++){
string str; cin >> str;
cout << findCount(str) << endl;;
}
return 0;
}