Remember the Word（trie树+DP）

E - Remember the Word

---

题目大意：

给定一个目标串和一本拥有n个单词的字典，求这些单词组成目标串有多少种组合方式。（结果对20071027取模）。

---

思路：

可将所有单词插入到字典树中，然后用目标串来进行匹配。这时可用dp数组来记录方案数。dp[i]表示以i开始的方案数，所以状态转换方程可为：
$$dp[i]=\sum dp[j](j>i)$$
通过枚举目标串的后缀来进行dp数组的更新，这样最后的答案就是以0开始的方案数（即dp[0])，在插入单词时对单词结尾进行标记，便于查找时对dp数组的更新，只有当出现结尾单词时才进行更新。

---

参考代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const ll M=20071027;
const int maxn=3e5+10;
const int maxm=4e5+10;
char a[maxn],b[110];
int trie[maxm][30],cnt[maxn],idx;
ll dp[maxn];
int n,len;
void insert(char a[])
{
	int p=0;
	for(int i=0;a[i];i++)
	{
		int x=a[i]-'a';
		if(!trie[p][x])trie[p][x]=++idx;
		p=trie[p][x];
	}
	cnt[p]++;
}
void query(int l)
{
	int p=0;
	for(int i=l;i<len;i++)
	{
		int x=a[i]-'a';
		if(!trie[p][x])return;
		p=trie[p][x];
		if(cnt[p]!=0)dp[l]=(dp[l]+dp[i+1])%M;
	}
}
int main()
{
	int s=0;
	while(~scanf("%s",a))
	{
		len=strlen(a);
		scanf("%d",&n);
		memset(trie,0,sizeof(trie));
		memset(cnt,0,sizeof(cnt));
		idx=0;
		for(int i=0;i<n;i++)
		{
			scanf("%s",b);
			insert(b);
		}
		memset(dp,0,sizeof(dp));
		dp[len]=1;
		for(int i=len-1;i>=0;i--)
			query(i);
		printf("Case %d: %lld\n",++s,dp[0]);

	}
	system("pause");
	return 0;
}