【问题描述】
给出一个由S个不同单词组成的字典和一个长字符串。把这个字符串分解成若干个单词的连接(单词可以重复使用),有多少种方法?
比如4个单词:{“a”,”b”,”cd”,”ab”},则”abcd”有两种分解方法:”a”+”b”+”cd” 或 “ab”+”cd”。
【输入格式】
第一行为长字符串(仅由小写字母组成)。 第二行一个整数n,表示字典包含的单词数。以下n行,每行为一个单词。
【输出格式】
输出长字符串的分解方案数除以 20071027 的余数。
【输入样例】
abcd 4 a b cd ab
【输出样例】
2
【样例解释】
时间限制:1秒 内存限制:64M
【数据范围】
n<=40000 长字符串长度不超过1 000 000,每个单词长度不超过100。
【来源】
大白书209页,LA3942
建立一棵trie树然后在原字符串上进行简单的动态规划就可以了。 d(i)完成了前i-1个字母有多少中情况。 以每一个字母为起点,在trie树上找有没有对应的单词,比如说i到j个字母为一个单词,d(j)=d(j)+d(i),这样就可以得出答案了。
详细代码如下:
#include<cstdlib> #include<cstdio> #include<cstring> #include<iostream> using namespace std; const int maxn=2000005; const int mod=20071027; int ch[maxn][26]={0},cnt=0,n,root=0,m; bool vis[maxn]={0}; char s[1000005],c[105]; int d[1000005]; void in()//建trie树 { int p=root,i=0,k; while(1) { if(c[i]==0) { vis[p]=1; return; } int k=c[i]-'a'; if(!ch[p][k]) ch[p][k]=++cnt; p=ch[p][k]; i++; } } void work(int i) { int p=root,k,j=i; while(i<=m) { if(vis[p])//找到了对应的单词 { d[i]=(d[i]+d[j])%mod; } int k=s[i]-'a'; p=ch[p][k]; if(!p) break;//一定要记得break,没有找的就跳出来 i++; } } int main() { freopen("in.txt","r",stdin); freopen("ou.txt","w",stdout); scanf("%s",s); scanf("%d",&n); for(int i=1;i<=n;i++) { scanf("%s",c); in(); } d[0]=1; m=strlen(s); for(int i=0;i<m;i++)//动规 { work(i); } cout<<d[m]; return 0; }