LA3942-Remember the Word（Trie）

题意：

有s个不同的单词，给出一个长字符串把这个字符串分解成若干个单词的连接（可重复使用），有多少种分解方法

分析：

dp[i]表示i开始的字符串能分解的方法数 dp[i]=sum(dp[i+len(x)]);单词x是i开始的字符串的前缀。

#include <map>
#include <set>
#include <list>
#include <cmath>
#include <queue>
#include <stack>
#include <cstdio>
#include <vector>
#include <string>
#include <cctype>
#include <complex>
#include <cassert>
#include <utility>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
typedef pair<int,int> PII;
typedef long long ll;
#define lson l,m,rt<<1
#define pi acos(-1.0)
#define rson m+1,r,rt<<11
#define All 1,N,1
#define read freopen("in.txt", "r", stdin)
#define maxnode 500000
const ll  INFll = 0x3f3f3f3f3f3f3f3fLL;
const int INF= 0x7ffffff;
const int mod =  ;
int dp[maxnode];
struct Trie{
    int ch[maxnode][];
    int val[maxnode];
    int num;
    void init(){num=;memset(ch[],,sizeof(ch[]));}
    int idx(char c){
        return c-'a';
    }
    void build(char *s,int v){
        int u=,n=strlen(s);
        for(int i=;i<n;++i){
            int c=idx(s[i]);
            if(!ch[u][c]){
                memset(ch[num],,sizeof(ch[num]));
                val[num]=;
                ch[u][c]=num++;
            }
            u=ch[u][c];
        }
        val[u]=v;
    }
    int query(char *s,int st){
        int u=,sum=,n=strlen(s);
        for(int i=st;i<n;++i)
        {
            int c=idx(s[i]);
            u=ch[u][c];
            if(u==)return sum;
            if(val[u]){
                sum+=dp[i+];
                sum%=mod;
            }
        }
        return sum;
    }
}trie;
int main()
{
    char str[],str1[];
    int m;
    while(~scanf("%s",str)){
        scanf("%d",&m);
        trie.init();
        for(int i=;i<m;++i){
            scanf("%s",str1);
            trie.build(str1,);
        }
        memset(dp,,sizeof(dp));
        int len=strlen(str);
        dp[len]=;
        for(int i=;i<=len;++i)
            dp[len-i]=trie.query(str,len-i);
            printf("%d\n",dp[]);
    }
return ;
}