HIHO Trie图 （ac自动机入门）

题目链接

Trie图，在trie树的基础上，使用kmp的思想，构造失配边形成trie图。
关键在于构造失配指针。简而言之，使用下面的规则即可

构造失败指针的过程概括起来就一句话：设这个节点上的字母为C， 沿着他父亲的失败指针走，直到走到一个节点，他的儿子中也有字母为C的节点。
然后把当前节点的失败指针指向那个字母也为C的儿子。 如果一直走到了root都没找到，那就把失败指针指向root

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<string>
#include<cmath>
#include<queue>
#include<map>
#include<set>
#include<cstdlib>
#include<vector>
using namespace std;
#define cl(a,b) memset(a,b,sizeof(a))
#define LL long long
#define pb push_back
#define gcd __gcd

#define For(i,j,k) for(int i=(j);i<k;i++)
#define lowbit(i) (i&(-i))
#define _(x) printf("%d\n",x)

const int maxn = 1e6+10;
const int inf  = 1 << 28;

/*

构造失败指针的过程概括起来就一句话：设这个节点上的字母为C，
沿着他父亲的失败指针走，直到走到一个节点，他的儿子中也有字母为C的节点。
然后把当前节点的失败指针指向那个字母也为C的儿子。
如果一直走到了root都没找到，那就把失败指针指向root

fail数组表示失配的指针
对于一个节点X，如果他不存在某个儿子，那么就直接在这里面存失配的指针
child[x][i]==-1 ==> child[x][i]=父亲节点的失配的指针指向的节点是i的指针

*/

struct AcTrie{
    int child[maxn][30];
    int fail[maxn];
    int ed[maxn];
    int root,L;

    int newNode(){
        cl(child[L],-1);
        fail[L]=ed[L]=0;
        return L++;
    }
    void init(){
        L = 0;
        root = newNode();
    }

    void insert(const char s[]){
        int len = strlen(s);
        int now = root;
        for(int i=0;i<len;i++){
            if(child[now][s[i]-'a']==-1){
                child[now][s[i]-'a']=newNode();
            }
            now = child[now][s[i]-'a'];
        }
        ed[now]++;
    }
    void build(){
        queue<int> q;
        fail[root]=root;
        for(int i=0;i<26;i++){
            if(child[root][i]==-1){
                child[root][i]=root;
            }
            else {
                fail[child[root][i]]=root;
                q.push(child[root][i]);
            }
        }
        while(!q.empty()){
            int now = q.front();q.pop();
            for(int i=0;i<26;i++){
                if(child[now][i]==-1){
                    child[now][i]=child[fail[now]][i];
                }
                else {
                    fail[child[now][i]]=child[fail[now]][i];
                    q.push(child[now][i]);
                }
            }
        }
    }

    int query(const char s[]){
        int len = strlen(s);
        int now = root;
        for(int i=0;i<len;i++){
            now = child[now][s[i]-'a'];
            if(ed[now])return 1;
        }
        return ed[now];
    }

}ac;

char s[maxn];

int main(){
    int n;
    while(~scanf("%d",&n)){
        ac.init();
        for(int i=0;i<n;i++){
            scanf("%s",s);
            ac.insert(s);
        }
        ac.build();
        scanf("%s",s);
        printf("%s\n",ac.query(s)?"YES":"NO");
    }
    return 0;
}