社区讨论

RE+TLE求助

P8306【模板】字典树参与者 4已保存回复 7

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
6 条
当前快照
1 份
快照标识符
@lzji0fxj
此快照首次捕获于
2024/08/07 14:58
2 年前
此快照最后确认于
2024/08/07 15:49
2 年前
查看原帖
CPP
#include<bits/stdc++.h>
using namespace std;
int n,Q;
string a[1000086];
string sb;
struct sbb{
	int tot,n=0;
}e[1000086][26+26+10];
int tot;
int maxx;
int now,ans;
int t;
int c;
int cti(char x){
	if(x>='a'&&x<='z')c=x-'a';
	else if(x>='A'&&x<='Z')c=x-'A'+26;
	else c=x-'0'+26+26;
	return c;
}
void add(string x){
	now=0;
	for(int i=0;i<x.size();i++){
		t=cti(x[i]);
		if(!e[now][t].tot)e[now][t].tot=++tot;
		e[now][t].n++;
		now=e[now][t].tot;
	}
	return;
}
int ask(string x){
	ans=0,now=0;
	for(int j=0;j<x.size();j++){
		t=cti(x[j]);
		if(!e[now][t].tot)break;
		if(j=x[x.size()-1]){
			ans+=e[now][t].n;
			break;
		}
		now=e[now][t].tot;
	}
	return ans;
}
void clean(){
	for(int i=0;i<=1000086;i++)for(int j=0;j<26+26+10;j++)e[i][j].n=0;
}
int _u,_v;
void run(){
	cin>>n>>Q;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		add(a[i]);
	}
	while(Q--){
		cin>>sb;
		cout<<ask(sb)<<endl;
	}
	clean();
	return;
}
int main(){
	int t1;
	cin>>t1;
	while(t1--)run();
}

回复

7 条回复,欢迎继续交流。

正在加载回复...