P2444 [POI2000]病毒 （AC自动机）

题目链接

题面：

题解：
如果在AC自动机上可以形成一个环且不经过危险标记，那么就可以。
在字典树上插入时，在每个序列最后打上危险标记。
很明显的是，如果一个节点的fail指针指向的节点有危险标记，那么这个点就应该有危险标记。
搜索时，ac自动机上不存在的点也应该搜索。

//爱的魔力转圈圈
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<cmath>
#include<queue>
#define ll long long
using namespace std;
const int maxn=30100;
int trie[maxn][2];
bool ed[maxn],ha[maxn],us[maxn];
int fail[maxn];
int cnt=0;

char str[maxn];

void _insert(void)
{
    int p=0;
    int k;
    int len=strlen(str);
    for(int i=0;i<len;i++)
    {
        k=str[i]-'0';
        if(!trie[p][k]) trie[p][k]=++cnt;
        p=trie[p][k];
    }
    ed[p]=true;

}

void getfail(void)
{
    queue<int>q;
    for(int i=0;i<2;i++)
    {
        if(trie[0][i])
        {
            fail[trie[0][i]]=0;
            q.push(trie[0][i]);
        }
    }
    while(q.size())
    {
        int now=q.front();
        q.pop();

        for(int i=0;i<2;i++)
        {
            int p=trie[now][i];
            if(p)
            {
                fail[p]=trie[fail[now]][i];
                if(ed[fail[p]]==true) ed[p]=true;
                q.push(p);
            }
            else
            {
                trie[now][i]=trie[fail[now]][i];
            }
        }
    }


}

bool dfs(int p)
{
    if(ha[p]) return true;
    if(us[p]||ed[p]) return false;
    ha[p]=us[p]=true;
    if(dfs(trie[p][0])) return true;
    if(dfs(trie[p][1])) return true;
    ha[p]=false;
    return false;
}

int main(void)
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",str);
        _insert();
    }
    getfail();
    if(dfs(0)) printf("TAK\n");
    else printf("NIE\n");
    return 0;
}