Trie LA3942

Trie的精髓在于利用了公共前缀，生成一个关系树，一条边代表一个字符，节点的val值来记录是否为单词结尾；

这次的数据结构Trie的所有操作直接写在了结构体里面，第一次这么写；

连续交了三遍Runtime Error，一个是Trie的节点数组开小了，总共有40万个节点，然后关键问题是我的d数组维护，我写的是d[state] = (d[state]+d[state+len[u]]) % MOD;不知道和现在的这个d[state] = (d[state]+d[i]) % MOD;区别在哪里；

然后最后query里我最后还有一句，就是刚好匹配到尾巴，要+1，不过感觉这句总可以写到里面去，放外面不太整齐；

维护数组还是尽量用循环变量的，用简单变量，避免那些奇奇怪怪的错误。

#include <iostream>
#include <fstream>
#include <algorithm>
#include <cstdio>
#include <cstring>

using namespace std;
#define MAXL 300005
#define MAXP 105
#define N 400005
#define SIG 26
#define MOD 20071027

int d[MAXL];
char s[MAXL], p[MAXP];
int Case=0, n;

struct Trie {
    int ch[N][SIG];
    bool val[N];
    int sz;

    Trie() { sz=1;memset(ch[0],0,sizeof(ch[0])); }
    void reset() {
        sz=1;memset(ch[0],0,sizeof(ch[0]));
    }
    int idx(char c) { return c-'a'; }
    //
    void Insert(char *s)
    {
        int u=0, n=strlen(s);
        for (int i=0;i<n;i++) {
            int c=idx(s[i]);
            if (!ch[u][c]) {
                //init
                memset(ch[sz],0,sizeof(ch[sz]));
                val[sz]=0;
                //record
                ch[u][c]=sz++;
            }
            u=ch[u][c];
        }
        val[u]=true; //len[u]=n;
    }

    void query(char *s,int state)
    {
        int u=0, n=strlen(s);
        for (int i=state;i<n;i++) {
            int c=idx(s[i]);
            if (val[u])
                d[state] = (d[state]+d[i]) % MOD;
            if (!ch[u][c]) return;
            u=ch[u][c];
        }
        if (val[u])
            d[state] = (d[state]+d[n]) % MOD;
    }
}tr;

int main()
{
   // freopen("1.in","r",stdin);

    while (scanf("%s",s)!=EOF) {
        tr.reset();
        scanf("%d",&n);
        for (int i=1;i<=n;i++) {
            scanf("%s",p);
            tr.Insert(p);
        }

        memset(d,0,sizeof(d));
        d[strlen(s)]=1;
        for (int i=strlen(s)-1;i>=0;i--) {
            tr.query(s,i);
        }
        printf("Case %d: %d\n",++Case,d[0]);
    }

    return 0;
}