BZOJ2938 || 洛谷P2444 [POI2000]病毒【AC自动机】

Time Limit: 1 Sec
Memory Limit: 128 MB

Description

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

Input

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

Output

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

---

题目分析

模式串很多但很短，文本串只有一个，由此想到AC自动机
那么简化题意，即能否构造一个字符串，满足将其送入AC自动机后沿着fail边永远也跳不到一个单词结点
再简化，即求AC自动机上是否存在一个不包含单词结点的环
显然如果存在，那么不断沿着这个环跳就能构造出无限长的安全串
构造AC自动机后dfs即可

注意dfs需要加入一个记录$u$是否已经作为环上可能节点dfs过的$used[]$数组，保证dfs效率
不加这个优化的话BZOJ上会T

---

#include<iostream>
#include<vector>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cstdio>
using namespace std;
typedef long long lt;
 
int read()
{
    int f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return f*x;
}
 
const int maxn=100010;
int n,cnt;
int ch[maxn][2],fail[maxn],rem[maxn];
int vis[maxn],used[maxn];
char pt[maxn];
queue<int> q;
 
void ins(char* ss,int len)
{
    int u=0;
    for(int i=0;i<len;++i)
    {
        int x=ss[i]-'0';
        if(!ch[u][x]) ch[u][x]=++cnt;
        u=ch[u][x];
    }
    rem[u]=1;
}
 
void ACM()
{
    for(int i=0;i<=1;++i)
    if(ch[0][i]) q.push(ch[0][i]);
     
    while(!q.empty())
    {
        int u=q.front(); q.pop();
        for(int i=0;i<=1;++i)
        {
            if(!ch[u][i]) ch[u][i]=ch[fail[u]][i];
            else{
                fail[ch[u][i]]=ch[fail[u]][i];
                if(rem[fail[ch[u][i]]]) rem[ch[u][i]]=1;
                q.push(ch[u][i]);
            }
        }
    }
}
 
int dfs(int u)
{
    vis[u]=1;
    for(int i=0;i<=1;++i)
    {
        int v=ch[u][i];
        if(vis[v]) return 1;
        if(used[v]||rem[v]) continue;
        used[v]=1;
        if(dfs(v)) return 1;
    }
    vis[u]=0;
    return 0;
}
 
int main()
{
    n=read();
    for(int i=1;i<=n;++i)
    {
        scanf("%s",&pt);
        ins(pt,strlen(pt));
    }
     
    ACM();
    if(dfs(0)) printf("TAK");
    else printf("NIE");
    return 0;
}