【模板】字典树luoguP8306

【模板】字典树

题目描述

给定 $n$ 个模式串 $s_1,s_2,\dots ,s_n$ 和 $q$ 次询问，每次询问给定一个文本串 $t_i$，请回答 $s_1\sim s_n$ 中有多少个字符串 $s_j$ 满足 $t_i$ 是 $s_j$ 的前缀。

一个字符串 $t$ 是 $s$ 的前缀当且仅当从 $s$ 的末尾删去若干个（可以为 0 个）连续的字符后与 $t$ 相同。

输入的字符串大小敏感。例如，字符串 Fusu 和字符串 fusu 不同。

输入格式

本题单测试点内有多组测试数据。

输入的第一行是一个整数，表示数据组数 $T$。

对于每组数据，格式如下：
第一行是两个整数，分别表示模式串的个数 $n$ 和询问的个数 $q$。
接下来 $n$ 行，每行一个字符串，表示一个模式串。
接下来 $q$ 行，每行一个字符串，表示一次询问。

输出格式

按照输入的顺序依次输出各测试数据的答案。
对于每次询问，输出一行一个整数表示答案。

样例 #1

样例输入 #1

3
3 3
fusufusu
fusu
anguei
fusu
anguei
kkksc
5 2
fusu
Fusu
AFakeFusu
afakefusu
fusuisnotfake
Fusu
fusu
1 1
998244353
9

样例输出 #1

2
1
0
1
2
1

提示

数据规模与约定

对于全部的测试点，保证 $1\le T,n,q\le 10^5$，且输入字符串的总长度不超过 $3\times 10^6$。输入的字符串只含大小写字母和数字，且不含空串。

说明

std 的 IO 使用的是关闭同步后的 cin/cout，本题不卡常。

#include<bits/stdc++.h>
using namespace std;
const int N=3e6+5;
int tot,t,n,q,tr[N][65],v[N];
string s;
int sw(char a)
{
if(a>='a'&&a<='z')
{
return a-'a'; 
}
if(a>='A'&&a<='Z')
{
return a-'A'+26;
}
return a-'0'+53;
}
void insert(string s)
{
int len=s.size();
int root=0;
for(int i=0;i<len;i++)
{
int j=sw(s[i]);
if(!tr[root][j])
{
tr[root][j]=++tot;
}
root=tr[root][j];
v[root]++;
}
}
int find(string s)
{
int len=s.size();
int root=0;
for(int i=0;i<len;i++)
{
int x=sw(s[i]);
if(!tr[root][x])return 0;
root=tr[root][x];
}
return v[root];
}
int main() {
ios::sync_with_stdio(0);
cin>>t;
while(t--)
{
for(int i=0;i<=tot;i++)
{
for(int j=0;j<=64;j++)
{
tr[i][j]=0;
}
}
for(int i=0;i<=tot;i++)
{
v[i]=0;
}
//memset(v,0,sizeof(v));
//memset(tr,0,sizeof(tr));
tot=0;
cin>>n>>q;
for(int i=1;i<=n;i++)
{
cin>>s;
insert(s);
}
for(int i=1;i<=q;i++)
{
cin>>s;
cout<<find(s)<<"\n";
}

}
}
/*
3
3 3
fusufusu
fusu
anguei
fusu
anguei
kkksc
5 2
fusu
Fusu
AFakeFusu
afakefusu
fusuisnotfake
Fusu
fusu
1 1
998244353
9
*/

原文地址：https://blog.csdn.net/no_play_no_games/article/details/143823310

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！