题目:http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=22109
题意:给出一个由S个不同单词组成的字典和一个长字符串。把这个字符串分解成若干个单词的连接(单词可以重复使用),由多少种方法?比如,有4个单词a,b,cd,ab,则abcd有两种分解方法:a+b+cd和ab+cd
分析:首先将输入的字典建成字典树。然后记忆化搜索~
代码:
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
const int maxn = 4e5+6;
const int mod =20071027;
int dp[maxn];
int son[maxn][26],cur,Len;
int end[maxn];
char ss[maxn],st[maxn];
int newnode()
{
end[cur]=0;
fill(son[cur],son[cur]+26,-1);
return cur++;
}
void Clear()
{
cur=0;
newnode();
}
void Insert(char str[])
{
int len=strlen(str),index,i,root=0;
for(i=0;i<len;i++)
{
index=str[i]-'a';
if(son[root][index]==-1)
son[root][index]=newnode();
root=son[root][index];
}
end[root]++;
}
int Query(int start)
{
if(start>=Len)
return 1;
if(dp[start]!=-1)
return dp[start];
int ret(0),index,i,root=0;
for(i=start;i<Len;i++)
{
index=ss[i]-'a';
if(son[root][index]!=-1 && end[son[root][index]])
ret+=Query(i+1)%mod;
if(son[root][index]==-1)
return dp[start]=ret%mod;
root=son[root][index];
}
return dp[start]=ret%mod;
}
int main()
{
int ncase=1,n;
while(scanf("%s%d",ss,&n)!=EOF)
{
Clear();
Len=strlen(ss);
fill(dp,dp+Len+1,-1);
while(n--)
{
scanf("%s",st);
Insert(st);
}
printf("Case %d: %d\n",ncase++,Query(0));
}
return 0;
}