字典树模板题。
ps:数组要开大,40w左右才行,不然疯狂re
代码:
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
int root;
int tot=1;
int trie[400500][30];
int cnt[400500];
void init()//最后清空,节省时间
{
memset(trie,0,sizeof(trie));
memset(cnt,0,sizeof(cnt));
tot=1;//RE有可能是这里的问题
}
void build_trie(char *s)
{
int len=strlen(s);
root=0;
for(int i=0;i<len;i++)
{
int id=s[i]-'a';
if(!trie[root][id])
trie[root][id]=++tot;
root=trie[root][id];
cnt[root]++;
}
}
int query(char *s)
{
int len=strlen(s);
root=0;
for(int i=0;i<len;i++)
{
int id=s[i]-'a';
if(!trie[root][id])
return false;
root=trie[root][id];
}
return cnt[root];
}
int main()
{
char s[105];
char t[105];
int ans=0;
init();
while(gets(s))
{
int x=strlen(s);
if(x==0)
break;
build_trie(s);
}
while(gets(t))
{
int flag=query(t);
printf("%d\n",flag);
}
}