BZOJ 2938: [Poi2000]病毒

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

题解

AC自动机+dfs。

如果一个串能无限长,也就是说它可以在AC自动机上一直进行匹配但就是匹配不上——也就是说匹配指针不能走到dg为1的结点。设某个dg为1的点为x，则自动机中：root..x是一个病毒串，那么“失败指针”指向x的节点y也不能走。

以上显然可得，接下来的做法比较奇怪（摘自hzwer.com）：方便起见我们直接把失败指针合并到子结点，dfs判断自动机上有没有环即可。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<cmath>
#include<algorithm>
#define MAXN 30002
using namespace std;
int n,q[MAXN],ins[MAXN],pd[MAXN];
int trie[MAXN][2],dg[MAXN],size=1,next[MAXN];
char ch[MAXN];
void insert()
{
     int l=strlen(ch),i,w=1,k;
     for(i=0;i<l;i++)
        {k=ch[i]-'0';
         if(!trie[w][k]) trie[w][k]=++size;
         w=trie[w][k];
        }
     dg[w]=1;
}
void init()
{
     scanf("%d",&n);
     int i;
     trie[0][0]=trie[0][1]=1;
     for(i=1;i<=n;i++)
        {scanf("%s",ch); insert();}
}
void getfail()
{
     int t=0,w=1,i,x,k;
     q[0]=1; next[1]=0; 
     while(t<w)
        {x=q[t]; t++;
         for(i=0;i<2;i++)
            {if(!trie[x][i])
                {trie[x][i]=trie[next[x]][i];
                 continue;
                }
             k=next[x];
             while(!trie[k][i]) k=next[k];
             next[trie[x][i]]=trie[k][i];
             if(dg[trie[k][i]]) dg[trie[x][i]]=1;
             q[w]=trie[x][i]; w++;
            }
        }
}
bool dfs(int x)
{
     ins[x]=1;//入栈 
     int i,k;
     for(i=0;i<2;i++)
        {k=trie[x][i];
         if(ins[k]) return true;//元素已在栈内则有环 
         if(dg[k]||pd[k]) continue;
         if(dfs(k)) return true;
         pd[k]=1;//表示以该结点为根的子树已判断过无环 
        }
     ins[x]=0;
     return false;
}
int main()
{
    init(); getfail();
    if(dfs(1)) puts("TAK");
    else puts("NIE");
    return 0;
}