hdu 2222 AC自动机入门

http://blog.csdn.net/niushuai666/article/details/7002823

这篇博客讲得不错

主要是要能理解这个trie图

有3个重要的指针，分别为p, p->fail, temp。

1.指针p，指向当前匹配的字符。若p指向root，表示当前匹配的字符序列为空。（root是Trie入口，没有实际含义）。

2.指针p->fail，p的失败指针，指向与字符p相同的结点，若没有，则指向root。

3.指针temp，测试指针 ，在建立fail指针时有寻找与p字符匹配的结点的作用。

详见代码注释

#include<bits/stdc++.h>
using namespace std;
const int N=10005;
const int maxlen=1000132;
const int maxn=10005*50;
int trie[maxn][26];
int fail[maxn];
int tag[maxn];
int sz;
queue<int >Q;

struct Aho
{
    int root=0;
    int newnode()//静态创建新节点
    {
        memset(trie[sz],-1,sizeof trie[sz]);
        tag[sz]=0;
        sz++;
        return sz-1;
    }
    void init()//初始化
    {
        sz=0;
        newnode();
    }
    void insert(char s[])   //插入字符串构建ac自动机，构建trie树
    {
        int len=strlen(s),p=0;;
        for (int i=0; i<len; i++)
        {
            int id=s[i]-'a';
            if (trie[p][id]==-1)
                trie[p][id]=newnode();
            p=trie[p][id];
        }
        tag[p]++;   //结束标记
    }
    void getfail() //构建自动机fail指针
    {
        while(!Q.empty()) Q.pop();
        fail[root]=root;            //root指向root
        for (int i=0; i<26; i++)
        {
            if (trie[root][i]==-1)//第一个字符不存在，指向root
                trie[root][i]=root;
            else    //第一个字符的fail指针指向root
            {
                fail[trie[root][i]]=root;
                Q.push(trie[root][i]);   //并放入队列，待bfs扩展
            }
        }
        while(!Q.empty())
        {
            int u=Q.front();    //取扩展节点
            Q.pop();
            for (int i=0; i<26; i++)//遍历所有子节点
            {
                if (trie[u][i]==-1)//如果不存在，则子节点直接指向fail[u]节点的对应子节点
                    trie[u][i]=trie[fail[u]][i];
                else     //如果存在，则该节点的fail指针指向fail[u]节点对应的子节点
                {
                    fail[trie[u][i]]=trie[fail[u]][i];
                    Q.push(trie[u][i]);     //继续扩展
                }
            }
        }
    }
} aho;

char ss[55];
char s[maxlen];

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        aho.init();
        int n;
        cin>>n;
        for (int i=0; i<n; i++)
        {
            scanf("%s",ss);
            aho.insert(ss);
        }
        aho.getfail();
        scanf("%s",s);
        int len=strlen(s);
        int p=aho.root;
        int ans=0;
        for (int i=0; i<len; i++)
        {
            int idx=s[i]-'a';
             p=trie[p][idx];
             int tmp=p; //取出当前字符在自动机上对应的节点
             while(tmp!=aho.root)//如果有fail节点则一直往上跳,直到跳回root表示没有了fail后继
              {
                  ans+=tag[tmp];//累计答案
                  tag[tmp]=0;//清空tag标记
                  tmp=fail[tmp];//跳到fail后继节点
              }
        }
        printf("%d\n",ans);

    }
    return 0;
}