hihoCoder week4 Trie图

ac自动机

题目链接 https://hihocoder.com/contest/hiho4/problem/1

参考：https://blog.csdn.net/baidu_30541191/article/details/47447175#

#include <iostream>
#include <cstring>
#include <string>
#include <queue>
using namespace std;

struct node
{
    node *nxt[], *fail;
    int ok;
    node() {
        ok = ;
        for(int i=; i<; i++) nxt[i]=NULL;
        fail=NULL;
    }
};

const int N = 1e6 + ;
char s[N];
node *root;

void ins(char *s, int len)
{
    node *rt = root;
    for(int i=; i<len; i++) {
        int x= s[i]-'a';
        if(rt->nxt[x] == NULL)
            rt->nxt[x] = new node();
        rt= rt->nxt[x];
    }
    rt->ok = ;
    return ;
}

void build()
{
    node *rt = root;
    rt->fail = NULL;

    queue<node*> que;
    que.push(rt);

    while(!que.empty()) {
        node *tmp = que.front();
        que.pop();
        for(int i=; i<; i++) {
            if(tmp->nxt[i] != NULL) {
                if(tmp == root)
                    tmp->nxt[i]->fail = root;
                else {
                    node *p = tmp->fail;
                    while(p!=NULL) {
                        if(p->nxt[i] != NULL) {
                            tmp->nxt[i]->fail = p->nxt[i];
                            break;
                        }
                        p = p->fail;
                    }
                    if(p==NULL) tmp->nxt[i]->fail = root;
                }
                que.push(tmp->nxt[i]);
            }
        }
    }
}

bool query(char *s, int len)
{
    node *rt = root;
    int cnt=;
    for(int i=; i<len; i++) {
        int x = s[i]-'a';
        while(!rt->nxt[x] && rt->fail != NULL)
            rt=rt->fail;
        rt=rt->nxt[x];
        if(!rt) rt = root;
        node * tmp = rt;
        while(tmp->ok==) {
            cnt++;
            tmp->ok = ;
            tmp = tmp->fail;
        }
    }
    if(cnt > )
        return true;
    return false;
}

int main()
{
    root = new node();
    int n; scanf("%d",&n);
    for(int i=; i<n; i++) {
        scanf("%s", s), ins(s, strlen(s));
    }

    build();
    scanf("%s", s);
    bool ok = query(s,strlen(s));
    if(ok)
        cout<<"YES"<<endl;
    else
        cout<<"NO"<<endl;
    return ;
}