LA3942 背单词(trie树&&简单动规）

【问题描述】

　　给出一个由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;}