題目
輸入的第一行爲一個正整數n,表示詞典的大小,其後n行,每一行一個單詞,單詞由不超過10個的小寫英文字母組成,可能存在相同的單詞,此時應將其視作不同的單詞。接下來的一行爲一個正整數m,表示詢問的次數,其後m行,每一行一個字符串,該字符串由不超過10個的小寫英文字母組成,表示一個詢問。對於每一個詢問,輸出一個整數Ans,表示詞典中以給出的字符串爲前綴的單詞的個數。
樣例輸入
5
babaab
babbbaaaa
abba
aaaaabaa
babaababb
5
babb
baabaaa
bab
bb
bbabbaab
樣例輸出
1
0
3
0
0
#include<iostream>
#include <string>
using namespace std;
typedef struct trie
{
char c;
int count;
struct trie* next[26];
} Trie;
int main()
{
Trie* root = new Trie;
for (unsigned i(0); i < 26; ++i)
root->next[i] = NULL;
int n, m;
string s;
cin >> n;
while (n--)
{
cin >> s;
Trie* p = root;
for (unsigned i(0); i < s.size(); ++i)
{
unsigned j = s[i] - 'a';
if (p->next[j])
{
++p->next[j]->count;
}
else
{
Trie* q = new Trie;
q->c = s[i];
q->count = 1;
for (unsigned k(0); k < 26; ++k)
q->next[k] = NULL;
p->next[j] = q;
}
p = p->next[j];
}
}
cin >> m;
while (m--)
{
cin >> s;
bool flag = false;
Trie* p = root;
for (unsigned i(0); i < s.size(); ++i)
{
unsigned j = s[i] - 'a';
if (!p->next[j])
{
flag = true;
break;
}
p = p->next[j];
}
if (flag) cout << '0' << endl;
else cout << p->count << endl;
}
return 0;
}