hdu 2222 DFA 模板题

真心对此题无奈了。。。。配得上2222这个称号了

题目要求我们算出一个字符串中包含着多少个他给出的单词。

坑爹的是这单词居然有一样的。。。。

如果一样的两个单词被给出，那么答案就应该是2，而不是1

样例如下：

Sample Input:

1

5

ab

ab

ab

ab

ab

ab

Sample Output:

5

有种被骗的赶脚~

#include
#include
#include
#include
using namespace std;
#define Maxn 6010000

struct nodes {
    nodes * next[26], *fail; //fail 指针指向该节点的最长后缀
    int no, inde;
} D[Maxn], *Q[Maxn];
int tot;

nodes *newnodes() {
    memset(D[tot].next, 0, sizeof (int) *26);
    D[tot].no = 0;
    D[tot].inde = tot;
    return &D[tot++];
}

int code(char c) {
    return c - 'a';
}

void insert(char s[], nodes *p, int inp) {
    for (int i = 0; s[i]; i++) {
        int tmp = code(s[i]);
        if (p->next[tmp]) {
            p = p->next[tmp];
        } else {
            p->next[tmp] = newnodes();
            p = p->next[tmp];
        }
    }
    p->no += inp;
}

void buildAC(nodes *p) {
    int L = 0, R = 0, i;
    for (i = 0; i < 26; i++) {
        if (p->next[i]) {
            Q[R++] = p->next[i];
            p->next[i]->fail = p;
        } else {
            p->next[i] = p;
        }
    }
    while (L < R) {
        p = Q[L++];
        for (i = 0; i < 26; i++) {
            if (p->next[i]) {
                p->next[i]->fail = p->fail->next[i]; //p->next[i]的fail是p的fail的next[i];
                Q[R++] = p->next[i];
            } else p->next[i] = p->fail->next[i]; //p->next[i]一定不为空.
        }
    }
}
char st[1000500];
bool mark[10505];

int main() {
    int N, i, cas;
    scanf("%d", &cas);
    while (cas--) {
        memset(mark, false, sizeof (mark));
        char str[90];
        scanf("%d", &N);
        tot = 0;
        nodes *root = newnodes();
        for (i = 0; i < N; i++) {
            scanf("%s", str);
            insert(str, root, 1);
            mark[i] = false;
        }
        buildAC(root);
        scanf("%s", st);
        nodes *p = root;
        nodes *g;
        int ans = 0;
        for (i = 0; st[i]; i++) {
            int tmp = code(st[i]);
            p = p->next[tmp];
            if (p->no != 0)
                ans += p->no;
            p->no = 0;
            g = p;
            while (g->inde != 0) {
                g = g->fail;
                if (g->no != 0)
                    ans += g->no;
                g->no = 0;
            }
        }
        printf("%d\n", ans);
    }
    return 0;
}