【Trie+dp】UVA1401 Remember the Word-优快云博客

本文介绍了一种利用字典树优化的字符串匹配算法，通过构建字典树存储短字符串集合，再利用字典树查找每个后缀的所有前缀，以此计算从i-L的方案数。该算法避免了暴力枚举的超时问题，适用于处理较长的字符串匹配问题。

这道题目由于S较为长，所以暴力枚举一定是超时的

所以考虑f[i]表示从i-L（后缀）的方案数

那么就可以推出 f[i]=sum{f[i+len(suf)],suf是i-L的一个前缀}

然后就可以在字典树上求这个前缀了

代码

#include<bits/stdc++.h>
using namespace std;
const int mod=20071027;
const int maxn=3e5+5;
char str[maxn],s[maxn];
int n,f[maxn],cas,ch[maxn][26],vis[maxn],cnt;
void insert(char *ss)
{
	int now=0;
	for(int k=0;ss[k];k++)
	{
		int to=ss[k]-'a';
		if(!ch[now][to]) ch[now][to]=++cnt;
		now=ch[now][to];
	}
	vis[now]=1;
}
int main()
{
	freopen("a.in","r",stdin);
	freopen("a.out","w",stdout);	
	while(scanf("%s",str)==1)
	{
		int len=strlen(str); cnt=0; cas++;
		memset(ch,0,sizeof(ch));
		memset(vis,0,sizeof(vis));
		scanf("%d",&n);
		for(int i=1;i<=n;i++)
		{
			scanf("%s",s);
			insert(s);
		}
		memset(f,0,sizeof(f));
		f[len]=1;
		for(int i=len-1;i>=0;i--)
		{
			int x=0;
			for(int j=i;j<len;j++)
			{
				int to=str[j]-'a';
				if(ch[x][to]==0) break;
				x=ch[x][to];
				if(vis[x])
					f[i]=(f[j+1]+f[i])%mod;
			}
		}
		printf("Case %d: %d\n",cas,f[0]);
	}
	return 0;
}