LA 3942 Remember the Word(Trie+DP)

题意：给出S个不同单词组成的字典和一个长字符串，把这个字符串分解成若干个由S组成的单词的连接有多少种方法。

思路：先把S都插入字典树，dp[i]表示从i开始的字符串的分解方式的个数，则每对应一个从i开始的在S集合中的串ss，dp[i]+=dp[i+strlen(ss)]。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<set>
#include<stack>
#include<cmath>
#include<vector>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-9
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
const int maxn=300000+10;
const int maxm=100+10;
const int maxnode=400000+10;
const int mod=20071027;
int ch[maxnode][26],val[maxnode],size;
char str[maxn],ss[maxm];
int dp[maxn];
void Init()
{
    memset(ch[0],0,sizeof(ch[0]));
    val[0]=0;size=0;
}
void insert(char *s)
{
    int u=0,n=strlen(s);
    for(int i=0;i<n;++i)
    {
        int c=s[i]-'a';
        if(!ch[u][c])
        {
            ch[u][c]=++size;
            memset(ch[size],0,sizeof(ch[size]));
            val[size]=0;
        }
        u=ch[u][c];
    }
    val[u]=n;
}
void find(int x,int n)
{
    int u=0;
    for(int i=x;i<n;++i)
    {
        int c=str[i]-'a';
        if(!ch[u][c]) break;
        if(val[ch[u][c]])
            dp[x]=(dp[x]+dp[i+1])%mod;
        u=ch[u][c];
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    int tcase=0;
    while(~scanf("%s",str))
    {
        tcase++;
        int m,n=strlen(str);
        scanf("%d",&m);
        Init();
        for(int i=0;i<m;++i)
        {
            scanf("%s",ss);
            insert(ss);
        }
        memset(dp,0,sizeof(dp));
        dp[n]=1;
        for(int i=n-1;i>=0;--i)
            find(i,n);
        printf("Case %d: %d\n",tcase,dp[0]);
    }
    return 0;
}