【BZOJ】2938 [POI2000]病毒（AC自动机）

题目

传送门：QWQ

传送到洛谷QWQ

分析

夏爷爷传送门

代码

#include <bits/stdc++.h>
using namespace std;
const int N=;
int n, son[N*][], fail[N*], end[N*], q[N*], vis[N*], c[N*];
char s[N][N*]; 

bool cycle(int u)
{
    c[u]=;
    for(int i=;i<=;i++)
    {
        int v=son[u][i];
        if(c[v]) return true;
        if(vis[v] || end[v]) continue;
        vis[v]=;
        if(cycle(v)) return true;
    }
    c[u]=; return false;
}

int find(int cur,int i)
{
    if(!cur) return ;
    if(son[cur][i]) return son[cur][i];
    return fail[son[cur][i]]=find(fail[cur],i);
}

int main()
{
    int newp=,rt=;
    scanf("%d",&n);
    for(int j=;j<=n;j++)
    {
        scanf("%s",s[j]+);
        int cur=rt, l=strlen(s[j]+);
        for(int i=;i<=l;i++)
        {
            if(!son[cur][s[j][i]-'']) son[cur][s[j][i]-'']=++newp;
            cur=son[cur][s[j][i]-''];
        }
        end[cur]=;
    }

    int l=,r=; q[]=;
    for(;l<=r;l++)
    {
        for(int i=;i<=;i++)
        if(son[q[l]][i])
        {
            int v=son[q[l]][i];
            fail[v]=find(fail[q[l]],i);
            end[v]|=end[fail[v]];
            q[++r]=son[q[l]][i];
        }
        else son[q[l]][i]=son[fail[q[l]]][i];//改儿子方便处理
    }

    if(cycle()) { printf("TAK\n");}
    else printf("NIE");
    return ;
}
/*
3
011
11
00000

*/