UVA-1401-dp+Trie

https://vjudge.net/problem/UVA-1401题目链接

 给出一个文本串和若干个单词，问由这些单词组成这个文本的不同方案个数

例 abc=a+b+c=ab+c=a+bc=abc  为他的所有可能的组合

  方程不难推出一个f[i]=SUM{f[j+1] | j>=i&&s[i,,j]是一个单词 } //其中f[i]表示s[i...sz]的方案个数，如果依次枚举判断这个前缀是不是一个单词的话复杂度达到O(sz^2)太大了，可以建一颗字典树每次从根出发沿着后面的单词走，遇到have为1表示单词存在累加答案否则跳过，走不下去时退出，由于深度最大100，复杂度就是O(100*sz);

#include<bits/stdc++.h>
using namespace std;
#define LL long long
const int MOD=20071027;
char s[300010];
int ch[200010][26];
bool have[300010];
int f[300010];
int size=0;
void insert(char *S)
{
    int n=strlen(S),u=0;
    for(int i=0;i<n;++i){
        int c=S[i]-'a';
        if(!ch[u][c]) ch[u][c]=size++;
        u=ch[u][c];
    }
    have[u]=1;
}
int main()
{
    char str[110];
    int cas=0;
    while(cin>>s+1){
        int ans=0;
        size=1;
        memset(ch,0,sizeof(ch));
        memset(have,0,sizeof(have));
        int sz=strlen(s+1);
        int n,i;
        cin>>n;
        for(i=1;i<=n;++i)
        {
            scanf("%s",str);
            insert(str);
        }
        f[sz+1]=1;
        for(i=sz;i>=1;--i)
        {
            f[i]=0;
            int j=i;
            int u=0;
            while(j<=sz&&ch[u][s[j]-'a']){
                u=ch[u][s[j]-'a'];
                if(have[u]) f[i]=(f[i]+f[j+1])%MOD;
                j++;
            }
        }
         //if(cas>=1)puts("");
        printf("Case %d: ",++cas);
        cout<<f[1]<<endl;
    }
    return 0;
}