P8306 字典树模板

【模板】字典树

题目描述

给定 \(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 \leq T, n, q\leq 10^5\)，且输入字符串的总长度不超过 \(3 \times 10^6\)。输入的字符串只含大小写字母和数字，且不含空串。

本质就是一颗多叉树

cnt[p]表示的是 p(包括p点)下面的字串(链)的个数 这样 我们每次插入一个字符串时 经过一个p就 cnt[p]++ 最后统计前缀子串个数就找到 字符串t的最后一个节点位置v 答案就是cnt[v]

注意最开始有个根节点0！ 所以清零要从0开始到上一次的idx

#include<bits/stdc++.h>
using namespace std;
const int N=3e6+5;
int tr[N][65],cnt[N];//26+26+10
int T,n,q;
int idx=0;
int getnum(char ch) {
	if(ch>='A'&&ch<='Z')return ch-'A'+1;
	if(ch>='a'&&ch<='z')return ch-'a'+1+26;
	if(ch>='0'&&ch<='9')return ch-'0'+1+26+26;
}
void insert(string s) {
	int l=s.size();
	int p=0;
	for(int i=0; i<l; i++) {
		int c=getnum(s[i]);
		if(!tr[p][c])
			tr[p][c]=++idx;
		p=tr[p][c];
		cnt[p]++;
	}
}
int query(string s) {
	int l=s.size();
	int p=0;
	for(int i=0; i<l; i++) {
		int c=getnum(s[i]);
		if(!tr[p][c])return 0;
		p=tr[p][c];
	}
	return cnt[p];
}
signed main() {
	ios::sync_with_stdio(false);
	cin>>T;
	while(T--) {
		for(int i=0; i<=idx; i++)
			for(int j=1; j<=65; j++)
				tr[i][j]=0;
		for(int i=0; i<=idx; i++)
			cnt[i]=0;
		idx=0;
		cin>>n>>q;
		for(int i=1; i<=n; i++) {
			string s;
			cin>>s;
			insert(s);
		}
		while(q--) {
			string s;
			cin>>s;
			cout<<query(s)<<"\n";
		}
	}
	return 0;
}