（trie）UVALive

题目链接

设d[i]表示从下标i的字符开始的字符串的分解方法数，显然有倒序的递推公式。

需要求每个位置开始是否能组成模式串的前缀，才可以建立正确的递推。

#include <iostream>
#include <string>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <queue>
#include <set>
#include <map>
#include <list>
#include <vector>
#include <stack>
#define mp make_pair
#define MIN(a,b) (a>b?b:a)
#define rank rankk
//#define MAX(a,b) (a>b?a:b)
typedef long long ll;
typedef unsigned long long ull;
const int MAX=3e5+5;
const int INF=1e9+5;
const int B=1024;//桶的大小
const double M=4e18;
const int MAX_NODE=MAX;
const int sigma_size=30;
using namespace std;
//const int MOD=1e9+7;
const int MOD=20071027;
typedef pair<int,int> pii;
const double eps=0.000000001;

struct Trie
{
    int ch[MAX_NODE][sigma_size];//点数、“字母”数
    int val[MAX_NODE];
    int num;//结点总数
    Trie(){num=1;memset(ch[0],0,sizeof(ch[0]));}//初始时仅有根节点
    void clear() { num = 1; memset(ch[0], 0, sizeof(ch[0])); }
    int idx(char c)//返回对应字符的编号
    {
        return c-'a';
    }
    /*
        插入字符串s,附加信息为v。注意v必须非0，因为0代表：本结点不是单词结点
    */
    void insert(const char *s,int v)
    {
        int u=0,len=strlen(s);
        for(int i=0;i<len;i++)
        {
            int c=idx(s[i]);
            if(!ch[u][c])//结点不存在
            {
                memset(ch[num],0,sizeof(ch[num]));
                val[num]=0;//中间节点的附加信息为0
                ch[u][c]=num++;//新建结点
            }
            u=ch[u][c];//往下走
        }
        val[u]=v;//字符串的最后一个字符的附加信息为v
    }
    /*
     查询字符串的“附加信息”
     查询过程中间中断返回0
    */
    int check(char *s)
    {
        int u=0,len=strlen(s);
        for(int i=0;i<len;i++)
        {
            int c=idx(s[i]);
            if(!ch[u][c])
                return 0;
            u=ch[u][c];
        }
        return val[u];
    }
    /*
    找字符串s的长度不超过len的前缀
    */
    void find_prefixes(const char *s,int len,vector <int> &ans)
    {
        int u=0;
        for(int i=0;i<len;i++)
        {
            if(s[i]=='')
                break;
            int c=idx(s[i]);
            if(!ch[u][c])
                break;
            u=ch[u][c];
            if(val[u]!=0)//过程中所有找到的全都push进去
                ans.push_back(val[u]);
        }
    }
};
char a[MAX];
int s,d[MAX],len,siz[MAX],cnt;
Trie dic;
char tem[MAX];
vector <int> an;
int main()
{
    while(~scanf("%s",a))
    {
        dic.clear();
        an.clear();
        scanf("%d",&s);
        for(int i=1;i<=s;i++)
        {
            scanf("%s",tem);
            siz[i]=strlen(tem);
            dic.insert(tem,i);
        }
        memset(d,0,sizeof(d));
        len=strlen(a);
        d[len]=1;
        for(int i=len-1;i>=0;i--)
        {
            an.clear();
            dic.find_prefixes(a+i,len-i,an);
            for(int j=0;j<an.size();j++)
            {
                d[i]=(d[i]+d[i+siz[an[j]]])%MOD;
            }
        }
        printf("Case %d: %d
",++cnt,d[0]);
    }
}