E - Remember the Word
题目大意:
给定一个目标串和一本拥有n个单词的字典,求这些单词组成目标串有多少种组合方式。(结果对20071027取模)。
思路:
可将所有单词插入到字典树中,然后用目标串来进行匹配。这时可用dp数组来记录方案数。dp[i]表示以i开始的方案数,所以状态转换方程可为:
d
p
[
i
]
=
∑
d
p
[
j
]
(
j
>
i
)
dp[i]=\sum dp[j] (j>i)
dp[i]=∑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;
}