|Hdu 2222|AC自动机|Keywords Search

Hdu传送门
AC自动机模板题，注意重复关键字的处理

#include<cstdio>    #include<algorithm>    #include<cstring>  #include<queue>   #define ms(i,j) memset(i,j, sizeof i);    using namespace std;const int MAXN = 10000 + 5, _SIZE = 26; int n;struct acam{    int sz;//结点编号     int res;    int ch[MAXN*51][_SIZE];//Tire     int last[MAXN*51];//后缀链接    int f[MAXN*51];//失配函数     int val[MAXN*51][2];//结点的权值     bool used[MAXN*51];//用过     void init()//初始化     {        ms(val,0);        ms(ch,0);        ms(used,false);        sz = 1;        res = 0;    }    void insert(char *s, int v)//插入一个模板     {        int u = 0;        int len = strlen(s);        for (int i=0;i<len;i++)        {            int c = s[i] - 'a';            if (ch[u][c])            {                u = ch[u][c];            } else            {                ch[u][c] = ++sz;                u = ch[u][c];            }            if (i==len-1)             {                val[u][0] = v;                val[u][1]++;            }        }    }    void g(int j)//递归更新cnt     {        if (j&&!used[val[j][0]])        {            res+=val[j][1];            used[val[j][0]] = true;            g(last[j]);        }    }     void find(char *T)//在T中匹配     {        int len = strlen(T);        int j = 0;        for (int i=0;i<len;i++)        {            int c = T[i] - 'a';            j = ch[j][c];            if (val[j][0]) g(j);            else if (last[j]) g(last[j]);        }    }     void getFail()//获得失配函数     {        queue<int> q;        f[0] = 0;        for (int c=0;c<_SIZE;c++)//初始化进队         {            int u = ch[0][c];            if (u)            {                q.push(u);                f[u] = 0;                last[u] = 0;            }         }        while (!q.empty())        {            int r = q.front(); q.pop();            for (int c=0;c<_SIZE;c++)            {                int u = ch[r][c];                if (!u)                {                    ch[r][c] = ch[f[r]][c];                    continue;                }                q.push(u);                int j = f[r];                while (j&&!ch[j][c]) j = f[j];                f[u] = ch[j][c];                last[u] = (val[f[u]][0]) ? (f[u]) : (last[f[u]]);             }        }    } }ac;char s[1000000 + 5];int main()   {         int kase;    scanf("%d", &kase);    while (kase--)    {        ac.init();        scanf("%d", &n);        for (int i=1;i<=n;i++)        {            scanf("%s", s);            ac.insert(s,i);        }        scanf("%s", s);        ac.getFail();        ac.find(s);        printf("%d\n", ac.res);    }    return 0;    }