LA、Remember the Word （字典树， 简单dp）

传送门

题意： 给你一个初始串 S，strlen(s) <= 3e5  然后给你 n 个单词。 n <= 4000,  每个单词的长度不超过 100 ；

　　　问你这个初始串，分割成若干个单词的连接的方案；（这些单词必须是给定的n个单词中的任意一个，一个单词可以被使用多次。）

解： 将 n 个单词建个字典树；

　　 dp[ i ] 表示，S的 0 ~ i - 1 切割成若干个 单词的方案数；

　　 枚举S， 枚举到 当前位置 i； 然后就在字典树找，以 S 的 i + 1 开始的， 能不能找到一个单词与之匹配；

　　 若能找到， 假设单词为 i + 1 ~ j； 那么就有 dp[ j + 1 ] = ( dp[ j + 1 ] + dp[ i ] ) % mod

　　  这只是个简单dp； 和字典树的简单应用的结合。

#include <bits/stdc++.h>
#define LL long long
#define rep(i, j, k) for(int i = j; i <= k; i++)
#define dep(i, j, k) for(int i = k; i >= j; i--)
#define INF 0x3f3f3f3f
#define inf 0x3f3f3f3f3f3f3f3f
#define mem(i, j) memset(i, j, sizeof(i))
#define pb push_back
using namespace std;

const int mod = ;

const int N = 3e5 + , M = 4e5 + ;
char s[N], b[];
int a[M][], tot, vis[N], dp[N];
int get(char x) {
    return x - 'a';
}
void join() {
    int now = ;
    rep(i, , (int)strlen(b) - ) {
        int id = get(b[i]);
        if(!a[now][id]) {
            mem(a[tot], ); vis[tot] = ;
            a[now][id] = tot++;
        }
        now = a[now][id];
    }
    vis[now] = ;
}
int main() {
    int cas = ;
    while(~scanf("%s", s)) {
        int n; scanf("%d", &n); tot = ;
        mem(a[], );
        rep(i, , n) {
            scanf("%s", b); join();
        }
        mem(dp, ); dp[] = ;
        rep(i, , (int)strlen(s) - ) {
            int now = ;
            rep(j, i, (int)strlen(s) - ) {
                int id = get(s[j]);
                if(!a[now][id]) break;
                now = a[now][id];
                if(vis[now]) {
                    dp[j + ] = (dp[j + ] + dp[i]) % mod;
                }
            }
        }
        printf("Case %d: ", ++cas);
        printf("%d\n", dp[(int)strlen(s)]);
    }
    return ;
}