Trie前缀树

最新推荐文章于 2024-05-21 23:00:30 发布

zzblac

最新推荐文章于 2024-05-21 23:00:30 发布

阅读量349

点赞数

分类专栏：模板整理

本文链接：https://blog.csdn.net/zzblac/article/details/38501591

版权

模板整理专栏收录该内容

4 篇文章 0 订阅

订阅专栏

前缀树

属于字符串，应该交给炮立方的，但是刷题遇到了所以简单看了一下

总之是建立一棵树，根节点为0，以小写字母为例，子节点是26个字母，每个节点的子节点还是字母，从跟节点向下走，组成字符串，每个节点上可以记录一些信息，比如这个字符串有没有出现，或者是一些其它的值。

比较容易实现，直接贴代码了，LA 3942 给出一个由S个不同单词组成的字典和一个厂字符串，把这个字符串分解成若干单调的连接（单词可重复使用），有多少种方法？

dp[i]=sum(dp[i-len(x)])

前缀树实现，数组大得惊人。。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define mod 20071027
int size,n,t[400005][26],dp[300005];
bool v[400005],visit[300005];
char word[4005][105],sequence[300005];
void insert(char *s)
{
	int id,node=0;
	for(int i=0;i<strlen(s);i++)
	{
		id=s[i]-'a';
		if(!t[node][id])
		{
			t[node][id]=size++;
		}
		node=t[node][id];
	}
	v[node]=1;
	return;
}
void dfs(int len,char *s)
{
	int node=0;
	dp[len]=0;
	for(int i=0;i<len;i++)
	{
		node=t[node][s[i]-'a'];
		if(node==0) return;
		if(v[node])
		{
			if(i<len-1)
			{
				if(!visit[len-i-1])
				{
					dfs(len-i-1,s+i+1);
					visit[len-i-1]=1;
				}
				dp[len]+=dp[len-i-1];
			}
			else dp[len]+=1;
			dp[len]%=mod;
		}
	}
	return;
}
void input()
{
	size=1;
	memset(v,0,sizeof(v));
	memset(t,0,sizeof(t));
	memset(visit,0,sizeof(visit));
	memset(dp,0,sizeof(dp));
	scanf("%d",&n);
	for(int i=0;i<=n-1;i++)
		scanf("%s",word[i]);
	return;
}
int main()
{
	int c=0,l;
	while(scanf("%s",sequence)!=EOF)
	{
		input();
		for(int i=0;i<n;i++)
			insert(word[i]);
		l=strlen(sequence);
		dfs(l,sequence);
		printf("Case %d: %d\n",++c,dp[l]);
	}
	return 0;
}

明天把cf455B做了巩固巩固，博弈的部分和dfs的部分都想好了，就是一开始不知道字典树，怎么琢磨都是要爆栈的节奏- -