社区讨论

求助trie模板

学术版参与者 4已保存回复 9

讨论操作

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

当前回复
9 条
当前快照
1 份
快照标识符
@lo7hsuxi
此快照首次捕获于
2023/10/27 02:03
2 年前
此快照最后确认于
2023/10/27 02:03
2 年前
查看原帖
CPP
#include<bits/stdc++.h>
using namespace std;
int Map[3000005][66],T,n,q,cur;
int exist[3000005];
int turn(char c)
{
    if(c>='A'&&c<='Z') return c-'A';
    else if(c>='a'&&c<='z') return c-'a'+26;
    else return c-'0'+52;
}
void insert(char s[])
{
	int root=0,len=strlen(s);
	for(int i=0;i<len;i++)
	{
		if(!Map[root][turn(s[i])]) Map[root][turn(s[i])]=++cur;
		root=Map[root][turn(s[i])];
		exist[root]++;
	}
}
int find(char s[])
{
	int root=0,len=strlen(s);
	for(int i=0;i<len;i++)
	{
		if(!Map[root][turn(s[i])]) return 0;
		root=Map[root][turn(s[i])];
	}
	return exist[root];
}
int main()
{
	cin>>T;
	for(int cnt=1;cnt<=T;cnt++)
	{
		memset(Map,0,sizeof(Map));
		memset(exist,false,sizeof(exist));
		cin>>n>>q;
		cur=0;
		char str[3141592];
		for(int i=1;i<=n;i++) scanf("%s",str),insert(str);
		for(int i=1;i<=q;i++) scanf("%s",str),cout<<find(str)<<endl;
	}
}
tle

回复

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

正在加载回复...