LA 3942 - Remember the Word

题目链接：https://icpcarchive.ecs.baylor.edu/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=1943

本题练习前缀树（Trie），又称字典树。

Trie结构体模板：

struct Trie
{
    int sz;
    int ch[maxnode][sigma_size];
    int val[maxnode];
    void init()
    {
        sz = 1;
        memset(ch[0],0,sizeof(ch[0]));
    }
    int idx(char c)
    {
        return c - 'a';
    }
    void insert(char *s,int v)
    {
        int u = 0;
        int n = strlen(s);
        for(int i=0;i<n;i++)
        {
            int c = idx(s[i]);
            if(!ch[u][c])
            {
                memset(ch[sz],0,sizeof(ch[sz]));
                val[sz] = 0;//注意这里不要写错
                ch[u][c] = sz++;
            }
            u = ch[u][c];
        }
        val[u] = v;
    }
    int find(char *s)
    {
        int u = 0;
        int n = strlen(s);
        for(int i=0;i<n;i++)
        {
            int c = idx(s[i]);
            if(!ch[u][c]) return 0;
            u = ch[u][c];
        }
        return val[u];
    }
};

对于本题而言，DP+Trie可以解决。

其中，d[i]代表以s[i-1]结尾的分解方案。

#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stdlib.h>
#include <time.h>
#include <vector>
#include <set>
#include <map>
#include <algorithm>

using namespace std;

#define maxnode 4010*100
#define sigma_size 26
#define MOD 20071027

struct Trie
{
    int sz;
    int ch[maxnode][sigma_size];
    int val[maxnode];
    void init()
    {
        sz = 1;
        memset(ch[0],0,sizeof(ch[0]));
    }
    int idx(char c)
    {
        return c - 'a';
    }
    void insert(char *s,int v)
    {
        int u = 0;
        int n = strlen(s);
        for(int i=0;i<n;i++)
        {
            int c = idx(s[i]);
            if(!ch[u][c])
            {
                memset(ch[sz],0,sizeof(ch[sz]));
                val[sz] = 0;//注意这里不要写错
                ch[u][c] = sz++;
            }
            u = ch[u][c];
        }
        val[u] = v;
    }
    int find(char *s)
    {
        int u = 0;
        int n = strlen(s);
        for(int i=0;i<n;i++)
        {
            int c = idx(s[i]);
            if(!ch[u][c]) return 0;
            u = ch[u][c];
        }
        return val[u];
    }
};
Trie trie;
char s[300005];
int d[300005];//d[i]代表以s[i-1]结尾的分解方案

int main()
{

#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
#endif
    int n;
    char temp[105];
    int cas = 0;
    while(scanf(" %s",s)!=EOF)
    {
        cas++;
        memset(d,0,sizeof(d));
        d[0] = 1;
        trie.init();
        scanf(" %d",&n);
        for(int i=0;i<n;i++)
        {
            scanf(" %s",temp);
            trie.insert(temp,1);
        }
        int len = strlen(s);
        for(int i=0;i<len;i++)
        {
            int u = 0;
            for(int j=i+1;j<=len;j++)
            {
                int c = trie.idx(s[j-1]);
                if(!trie.ch[u][c]) break;
                if(trie.val[trie.ch[u][c]])
                {
                    d[j] += d[i];
                    if(d[j]>=MOD) d[j] -= MOD;
                }
                u = trie.ch[u][c];
            }
        }
        printf("Case %d: %d\n",cas,d[len]);
    }
}