题意:给一个由s个单词组成的字典和一个长字符串。把这个字符串分解成若干个单词的连接(单词可重复使用),有多少种方法?
思路:令d(x)表示从字符i开始的字符串(即后缀s[i....L])的分解方案数,则d(i)=sum{d+len(x) | 单词x是s[i...L]的前缀}。
所以可以先把单词写入trie树,然后从从后往前在长串的每个位置找前缀即可。
#include<iostream>
#include<stdio.h>
#include<math.h>
#include <string>
#include<string.h>
#include<map>
#include<queue>
#include<set>
#include<utility>
#include<vector>
#include<algorithm>
#include<stdlib.h>
using namespace std;
#define eps 1e-8
#define pii pair<int,int>
#define inf 0x3f3f3f3f
#define rd(x) scanf("%d",&x)
#define rd2(x,y) scanf("%d%d",&x,&y)
#define rd3(x,y,z) scanf("%d%d%d",&x,&y,&z)
#define mo(x) memset(x,0,sizeof(x))
#define ll long long int
#define ma(x,y) (x)>(y)?(x):(y)
#define mi(x,y) (x)<(y)?(x):(y)
#define mod 20071027
#define maxn 500010
#define maxm 10000001
struct Trie{
int next[26];
int k;
}trie[maxn];
int tot;
int newnode(){
for(int i=0;i<26;i++) trie[tot].next[i]=-1;
trie[tot].k=0;
tot++;
return tot-1;
}
void inser(char str[]){
int now=0;
int len=strlen(str);
for(int i=0;i<len;i++){
if(trie[now].next[str[i]-'a']==-1)
trie[now].next[str[i]-'a']=newnode();
now=trie[now].next[str[i]-'a'];
}
trie[now].k=1;
}
char str[maxn];
char s[200];
int len,n,d[maxn];
int main()
{
int t=0;
while(~scanf("%s",str)){
rd(n);
tot=0;
newnode();
for(int i=1;i<=n;i++){
scanf("%s",s);
inser(s);
}
len=strlen(str);
memset(d,0,sizeof(d));
d[len]=1;
for(int i=len-1;i>=0;i--){
int now=0;
for(int j=i;j<len;j++){
if(trie[now].next[str[j]-'a']==-1) break;
now=trie[now].next[str[j]-'a'];
if(trie[now].k) d[i]=(d[i]+d[j+1])%20071027;
}
}
printf("Case %d: %d\n",++t,d[0]);
}
return 0;
}