FF得到了一个有n个数字的集合。不要问我为什么,有钱,任性。
FF很好奇的想知道,对于数字x,集合中有多少个数字可以在x前面添加任意数字得到。
如,x = 123,则在x前面添加数字可以得到4123,5123等。
Input
多组输入。
对于每组数据
首先输入n(1<= n <= 100000)。
接下来n行。每行一个数字y(1 <= y <= 100000)代表集合中的元素。
接下来一行输入m(1 <= m <= 100000),代表有m次询问。
接下来的m行。
每行一个正整数x(1 <= x <= 100000)。
Output
对于每组数据,输出一个数字代表答案。
Sample Input
3
12345
66666
12356
3
45
12345
356
Sample Output
1
0
1
思路:
倒着建立字典树就行了,并且用个数组来标记每个节点被经过的次数
#include<bits/stdc++.h>
using namespace std;
const int maxx=1e5+10;
int trie[maxx][11];
int num,c[maxx];//c来标记每个节点被经过的次数
int fin(int root,char a[])
{
int len=strlen(a),n,i;
for(int i=len-1; i>=0; i--)
{
n=a[i]-'0';
if(trie[root][n]==0)
return 0;
root=trie[root][n];
}
return c[root];
}
void creat(int root,char a[])
{
int len=strlen(a);
int n;
for(int i=len-1; i>=0; i--)
{
n=a[i]-'0';
if(trie[root][n]==0)
trie[root][n]=++num;
c[root]++;
root=trie[root][n];
}
}
int main()
{
std::ios::sync_with_stdio(false);
int n,m,i;
char a[20];
while(cin>>n)
{
memset(trie,0,sizeof(trie));
num=0;
memset(c,0,sizeof(c));
for(i=0; i<n; i++)
{
cin>>a;
creat(0,a);
}
cin>>m;
for(i=0; i<m; i++)
{
cin>>a;
cout<<fin(0,a)<<endl;
}
}
return 0;
}