题目描述
给定n个数字的集合,对于数字x,集合中有多少个数字在x前面添加任意数字可以得到。
解题思路
首先先将输入的字符逆序,这样就可以转化为前缀的问题。
由测试数据可以看出,当要查询的x为集合中的某一数字时,是不进行计数的。
将数字逆序构建字典树,cnt用来计算字符在字典树特定位置中出现的次数,在最后的位置cnt需要减去1,因为完整的字符不符合。当要查询一个前缀的时候,只需要输出他最后一个字符的cnt即可。
代码部分
#include <bits/stdc++.h>
using namespace std;
const int en = 10;
const int maxn = 1e6;
struct Trie
{
int cnt;
Trie *nex[en];
} T[maxn];
int top;
inline int idx(char c)
{
return c - '0';
}
inline Trie * CreatTrie()
{
Trie *p = &T[top ++];
p -> cnt = 0;
for(int i = 0; i< en; ++ i) p -> nex[i] = NULL;
return p;
}
inline Trie * Init()
{
top = 0;
return CreatTrie();
}
void Insert(Trie *root, string s)
{
Trie *p = root;
for(int i = 0; i < s.size(); ++ i)
{
int temp = idx(s[i]);
if(p -> nex[temp] == NULL) p -> nex[temp] = CreatTrie();
p = p -> nex[temp];
p -> cnt ++;
}
p -> cnt --;
}
int Search(Trie *root, string s)
{
Trie *p = root;
for(int i = 0; i < s.size(); ++ i)
{
int temp = idx(s[i]);
if(p -> nex[temp] == NULL) return 0;
p = p -> nex[temp];
}
return p -> cnt;
}
int main()
{
int n, m;
string s;
while(cin >> n)
{
Trie *root = Init();
for(int i = 1; i <= n; ++ i) cin >> s, reverse(s.begin(), s.end()), Insert(root, s);
cin >> m;
for(int i = 1; i <= m; ++ i) cin >> s, reverse(s.begin(), s.end()), cout<< Search(root, s) << endl;
}
return 0;
}