HDU 2222 Keywords Search

　　第一次写AC自动机，参考集训手册题解的代码，但是还是花了三个小时的时间才A掉。字符串

插入字典树的过程就是一般字典树形成的过程。这里关键在于求失败指针。有点像KMP求next的过程。

　　对于每个结点，我们可以这样处理：设这个结点上的字母为C，沿着他父亲的失败指针走，直到

走到一个结点，他的儿子中也有字母为C的节点。然后把当前结点的失败指针指向那个字母也为C的结

点。如果一直走到了root都没找到，那就把失败指针指向root。最开始，我们把root加入队列(root

的失败指针显然指向自己)，这以后我们每处理一个结点，就将他的所有儿子几点入队。

　　假设有一个节点k，他的失败指针指向j。那么k,j满足这个性质：设root到j的距离为n，则从k之上

的第n - 1个节点到k所组成的长度为n的单词，与从root之后第一个结点到j所组成的单词相同。

/*Accepted    2222    187MS    26636K    2048 B    G++    Yu*/

#include<stdio.h>

#include<string.h>

#include<stdlib.h>

#include<queue>

using namespace std;

#define init(x) (memset(trie[x], 0, sizeof(trie[x])))

#define next 26

#define root 0

#define cnt 27



const int MAXN = 10001;

const int MAXL = 1000001;

const int MAXM = 28;

int trie[MAXN * 50][MAXM], ntp;

char src[MAXL];

queue<int> q;



void inser(char *s)

{

    int i, cur = root, t;

    for(i = 0; s[i]; i ++)

    {

        t = s[i] - 'a';

        if(!trie[cur][t])

            init(ntp),trie[cur][t] = ntp ++;

        cur = trie[cur][t];

    }

    ++ trie[cur][cnt];

}



void GetNext() //广搜的过程

{

    int cur, t, i;

    q.push(root);

    trie[root][next] = root;

    while(!q.empty())

    {

        cur = q.front(), q.pop();

        for(i = 0; i < 26; i ++)

        {

            if(trie[cur][i])

            {

                if(cur == root) //如果父亲结点是根，那么失败就只能返回根

                    trie[trie[cur][i]][next] = root;

                else

                {

                    t = trie[cur][next]; //trie[cur][i]的父亲cur的失败指针

                    while(t && !trie[t][i]) //找到一个和trie[cur][i]字符相同的结点所处的位置

                        t = trie[t][next];

                    trie[trie[cur][i]][next] = trie[t][i]; //将trie[cur][i]的失败指针指向相同的点

                }

                q.push(trie[cur][i]); //将cur的所有儿子结点入队

            }

            else {

                trie[cur][i] = trie[trie[cur][next]][i]; //不存在这一点，指向父亲结点失败指针的第i个儿子

            }

        }

    }

}



int cal(char *s)

{

    int i, j, cur = root, t, ans = 0;

    for(i = 0; s[i]; i ++)

    {

        t = s[i] - 'a';

        cur = trie[cur][t];

        j = cur;

        while(j && trie[j][cnt] != -1)

        {

            ans += trie[j][cnt];

            trie[j][cnt] = -1;

            j = trie[j][next];

        }

    }

    return ans;

}



int main()

{

    int T, n;

    scanf("%d", &T);

    while(T --)

    {

        scanf("%d", &n);

        ntp = 1;

        init(root);

        while(n --)

        {

            scanf("%s", src);

            inser(src);

        }

        GetNext();

        scanf("%s", src);

        printf("%d\n", cal(src));

    }

    return 0;

}