题目传送门
当你去购物时,你可以在仓库里通过电脑和互联网搜索有用的商品。首先,你给搜索系统一个关于某物的名称,然后系统用结果进行响应。现在您在存储库中获得了许多商品名称和一些查询,并且需要模拟这个过程。
输入
只有一种情况。首先,有一个整数P (1<=P<=10000)表示存储库中商船名的数量。接下来的P行每个都包含一个字符串(它的长度不超过20,并且所有的字母都是小写的)。然后有一个整数Q(1<=Q<=100000)表示查询的数量。下一个Q行每个都包含一个字符串(与前面的描述相同的限制)作为搜索条件。
输出
对于每个查询,您只需输出商品的数量,商品名称包含搜索字符串作为其子字符串。
Sample Input
20
ad
ae
af
ag
ah
ai
aj
ak
al
ads
add
ade
adf
adg
adh
adi
adj
adk
adl
aes
5
b
a
d
ad
s
Sample Output
0
20
11
11
2
字典树是用来查找前缀的,而这个是判断子串,故将一个字符串拆分 例如abcde 拆分成 abcde bcde cde de e
注意aabab拆成 aabab aaba aab ab b存在一个字符串里多个串前缀相同
(前四个都有共同的前缀a)
所以搞一个id 判断一下这个串的最后操作者与目前操作者一样不一样,一样的话不需要计算,不一样cnt++;串的中间cnt++则是前缀的。。需要这样
#include<cstdio>
#include<cstring>
using namespace std;
struct trie
{
int id;
int cnt;
trie *nex[26];
trie()
{
id=-1;
cnt=0;
for(int i=0;i<26;i++)
nex[i]=NULL;
}
};
trie *root;
void create(char *str,int p,int num)
{
int len=strlen(str);
trie *a=root;
for(int i=p;i<len;i++)
{
int ch=str[i]-'a';
if(a->nex[ch]==NULL)
a->nex[ch]=new trie();
a=a->nex[ch];
if(a->id!=num)
{
a->id=num;
a->cnt++;
}
}
}
int seek(char *str)
{
int len=strlen(str);
trie *a=root;
for(int i=0;i<len;i++)
{
int ch=str[i]-'a'; //能不能不要在写错啊 'a' 、'0' 明显不一样~~
a=a->nex[ch];
if(a==NULL)
return 0;
}
return a->cnt;
}
void del(trie *a)
{
for(int i=0;i<26;i++)
if(a->nex[i])
del(a->nex[i]);
delete(a);
}
char s[22];
int main()
{
int n;
scanf("%d",&n);
root=new trie();
for(int i=1;i<=n;i++)
{
scanf("%s",s);
int len=strlen(s);
for(int j=0;j<len;j++) //将串拆分
create(s,j,i);
}
scanf("%d",&n);
while(n--)
{
scanf("%s",s);
printf("%d\n",seek(s));
}
del(root);
}