bzoj2938 [Poi2000]病毒 AC自动机+DP

Description

二进制病毒审查委员会最近发现了如下的规律：某些确定的二进制串是病毒的代码。如果某段代码中不存在任何一段病毒代码，那么我们就称这段代码是安全的。现在委员会已经找出了所有的病毒代码段，试问，是否存在一个无限长的安全的二进制代码。
示例：
例如如果{011, 11, 00000}为病毒代码段，那么一个可能的无限长安全代码就是010101…。如果{01, 11, 000000}为病毒代码段，那么就不存在一个无限长的安全代码。
任务：
请写一个程序：
l 读入病毒代码；
l 判断是否存在一个无限长的安全代码；
l 将结果输出
Input

第一行包括一个整数n，表示病毒代码段的数目。以下的n行每一行都包括一个非空的01字符串——就是一个病毒代码段。所有病毒代码段的总长度不超过30000。
Output

你应在在文本文件WIN.OUT的第一行输出一个单词：
l TAK——假如存在这样的代码；
l NIE——如果不存在。
Sample Input

3
01
11
00000
Sample Output

NIE

题解：我一看这个和那个bzoj1030有点像，都是求不合法，但是不是方案，是一个可能合法的串，但是打了半天WA，结果发现是要无限长度的，，毕竟AC自动机做的少。瞬间就虚了。。果然还是套路不够深，膜了一发题解，发现是用dfs。。每次dfs不走重复的路径，不走字串的结束点，然后判断是否有环就行了。
注意这里的自动机和以往不同，被坑死。。

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<queue>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
typedef long long ll;
const int N=4e5+5;
int val[N],fail[N],trie[N][3],vis[N],ins[N],tot,n,m;
char ch[N];
int q[N];
inline void insert()
{
    int len=strlen(ch),x=0;
    fo(i,0,len-1)
    {
        int c=ch[i]-'0';
        if (trie[x][c])x=trie[x][c];
        else x=trie[x][c]=++tot;
    }
    val[x]=1;
}
inline void getfail()
{
    int t=0,w=0;
    fo(i,0,1)if (trie[0][i])q[w++]=trie[0][i];
    while (t<w)
    {
        int x=q[t++];
        fo(i,0,1)
        if (trie[x][i])
        {
            int y=trie[x][i],j=fail[x];
            while(j&&!trie[j][i])j=fail[j];
            fail[trie[x][i]]=trie[j][i];
            if (val[trie[j][i]])val[trie[x][i]]=1;
            q[w++]=trie[x][i];
        }
        else trie[x][i]=trie[fail[x]][i];//这是多加的一句。
    }
}
bool dfs(int x)
{
    if (x)ins[x]=1;
    fo(i,0,1)
    {
        int v=trie[x][i];
        if (ins[v])return 1;
        if (vis[v]||val[v])continue;
        vis[v]=1;
        if (dfs(v))return 1;
    }
    ins[x]=0;
    return 0;
}
int main()
{
    scanf("%d",&n);
    fo(i,1,n)
    {
        scanf("%s",ch);
        insert();
    }
    getfail();
    if (dfs(0))printf("TAK");
    else printf("NIE");
    return 0;
}