版权声明:本文为博主原创文章,转载请说明出处。 https://blog.csdn.net/xianpingping/article/details/83096900
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn=2000000+10;
const int inf=0x3f3f3f3f;
int tree[maxn][30];
bool flagg[maxn];
int sum[maxn];///有时要加这个数组,目的是为了统计以该节点为前缀的单词的数量
int tot;
void insert_(string str){
int len=str.size();
int root=0;
for(int i=0;i<len;i++){
int id=str[i]-'a';
if(!tree[root][id]){
tree[root][id]=++tot;
}
root=tree[root][id];
sum[root]++;
}
flagg[root]=true;
}
int find_(string str){
int len=str.size();
int root=0;
for(int i=0;i<len;i++){
int id=str[i]-'a';
if(!tree[root][id])
return 0;///该单词没有出现过
root=tree[root][id];
}
/*if(flagg[root])
return sum[root];///出现过,不再算贡献
else
return 0;///没有出现过
*/
return sum[root];
}
char ss[maxn];
void init(){
memset(sum,0,sizeof(sum));
tot=0;
/// memset(flagg,0,flagg);
}
int main()
{
init();
int cnt=0;
while(gets(ss)){
if(ss[0]=='\0')
break;
insert_(ss);
}
///输入空行后。
while(scanf("%s",ss)!=EOF){
printf("%d\n",find_(ss));
}
return 0;
}