洛谷-P8306模板字典树

最新推荐文章于 2025-12-01 22:45:36 发布

原创最新推荐文章于 2025-12-01 22:45:36 发布 · 195 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#算法

该文描述了一种使用Trie树数据结构处理字符串查询的方法。通过构建Trie树对模式串进行预处理，可以高效地判断给定字符串是否是模式串的前缀，并计算满足条件的前缀个数。在每次查询时，若字符串中的字符在树中不存在，立即返回0；否则按照字符路径更新p值，最后返回终点的计数，表示前缀个数。代码示例展示了如何插入字符串到Trie树以及执行查询操作。

将模式串进行建树，每个字符出现的时候都进行标记，即red[p]++;全部建立完毕后

对于每次询问，如果该字符串的某个字符没有出现过，直接结束，不满足该字符串为模式串的前缀，如果出现过，那么p值仍然等于trie[p][x]，直到遍历到最后一个字符位置时，返回最后一个字符出现的次数，即为前缀个数。

int find(string s)
{
	int p=0;
	int sum=0;
	for(int i=0;i<s.size();i++)
	{
		int x=s[i]-'0';
		if(!trie[p][x])
		return 0;
		p=trie[p][x];	
	}
	return red[p];
}

由于对于每次询问，p的值都是从0开始，例如在查找fusu满足多少个字符串的前缀时，akakefusu并不满足，因为fusu中的最后一个字符u与akakefusu中的u的位置不同，也即p的值不同，所以这个不满足前缀。

用memset会MLE

 for(int i=0;i<=id;i++)
{
	for(int j=0;j<=100;j++)
	{
		 trie[i][j]=0;
    }
}
for(int i=0;i<=id;i++)
red[i]=0;

代码：

#include<iostream>
#include<cstring>
using namespace std;
int trie[3000005][100];
int red[3000005];
int id;
int find(string s)
{
	int p=0;
	int sum=0;
	for(int i=0;i<s.size();i++)
	{
		int x=s[i]-'0';
		if(!trie[p][x])
		return 0;
		p=trie[p][x];	
	}
	return red[p];
}
void insert(string s)
{
	int p=0;
	for(int i=0;i<s.size();i++)
	{
		int x=s[i]-'0';
		if(trie[p][x]==0)
		trie[p][x]=++id;
		p=trie[p][x];//记录每个字符出现在值p的位置
		red[p]++; //记录每个字符出现在相同位置时的次数
	}
}
int main()
{
	    int t;
	    cin>>t;
	    while(t--)
	    {
		    int n,m;
		    cin>>n>>m;
		    for(int i=0;i<=id;i++)
		    {
			    for(int j=0;j<=100;j++)
			    {
				    trie[i][j]=0;
			    }
		    }
		    for(int i=0;i<=id;i++)
		    red[i]=0;
		    string s;
		    id=0;
		    for(int i=0;i<n;i++)
		    {
			    cin>>s;
			    insert(s);
		    }
		    for(int i=0;i<m;i++)
		    {
			    cin>>s;
			    cout<<find(s)<<endl;
		    }
	    }
	        return 0;
}