输入
输入的第一行为一个正整数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给出的字符串为前缀的单词的个数。
思路:
字典树模版,维护一棵26叉树,遍历每个单词的每个字母进行建树。
#include <iostream>
#include <cstring>
#include <cstdio>
#include <string>
using namespace std;
const int maxn = 30;
typedef struct Trie
{
Trie *next[maxn]; //下一层的节点数(小写字母个数)
int v; //从根节点到当前节点为前缀的单词个数
Trie(){}
};
Trie root;
void creatTrie(string s)
{
int len = s.length();
Trie *p = &root;
Trie *q;
for (int i = 0; i < len; ++i) { //对单词s建立字典树
int id = s[i] - 'a';
if (p->next[id] == nullptr) {
q = new Trie();
for (int j = 0; j < maxn; ++j)
q->next[j] = nullptr;
q->v = 1;
p->next[id] = q;
}
else {
p->next[id]->v++;
}
p = p->next[id];
}
}
int findTrie(string s) //查找
{
int len = s.length();
auto *p = &root;
for (int i = 0; i < len; ++i) {
int id = s[i] - 'a';
if (p->next[id] == nullptr) return 0;
p = p->next[id];
}
return p->v;
}
int main()
{
int n,m;
cin >> n;
string s;
for (int i = 0; i < n; ++i) {
cin >> s;
creatTrie(s);
}
cin >> m;
for (int i = 0; i < m; ++i) {
cin >> s;
int res = findTrie(s);
cout << res << endl;
}
return 0;
}