BZOJ [Poi2000]病毒 AC自动机 + DFS + 细节

题目描述

二进制病毒审查委员会最近发现了如下的规律：某些确定的二进制串是病毒的代码。如果某段代码中不存在任何一段病毒代码，那么我们就称这段代码是安全的。现在委员会已经找出了所有的病毒代码段，试问，是否存在一个无限长的安全的二进制代码。

示例：

例如如果{011, 11, 00000}为病毒代码段，那么一个可能的无限长安全代码就是010101…。如果{01, 11, 000000}为病毒代码段，那么就不存在一个无限长的安全代码。

任务：

请写一个程序：

1.在文本文件WIR.IN中读入病毒代码；

2.判断是否存在一个无限长的安全代码；

3.将结果输出到文件WIR.OUT中。

输入输出格式

输入格式：

在文本文件WIR.IN的第一行包括一个整数n

输出格式：

在文本文件WIR.OUT的第一行输出一个单词：

TAK——假如存在这样的代码；

NIE——如果不存在。

题解：

求是否能构造一个无限长的 01 串，使得串不包含任意一个提前给定的子串.
建出 AC 自动机，显然不能走到打上终止标记的节点.
每次遍历 $ch[u][0]$， $ch[u][1]$ 找环即可.
找不到环则说明找不到长度为无限长的 01 串，否则可以找到长度为无限长的 01 串.

#include<bits/stdc++.h>
#define setIO(s) freopen(s".in","r",stdin) 
#define maxn 40000
#define N 2 
using namespace std;
char str[maxn]; 
queue<int>Q; 
int ch[maxn][3], tag[maxn], f[maxn], vis[maxn], done[maxn];  
int tot,root=0; 
void ins(char p[])
{
    int n=strlen(p+1),cur=0; 
    for(int i=1;i<=n;++i)
    {
        int c=p[i]-'0';  
        if(!ch[cur][c]) ch[cur][c]=++tot; 
        if(tag[cur]) tag[ch[cur][c]]=1; 
        cur=ch[cur][c]; 
    } 
    tag[cur]=1;       
} 
void build()
{
    for(int i=0;i<2;++i) if(ch[root][i]) Q.push(ch[root][i]); 
    while(!Q.empty())
    {
        int u=Q.front();Q.pop(); 
        for(int i=0;i<N;++i)
        {
            int q=ch[u][i]; 
            if(!q) 
            {
                ch[u][i]=ch[f[u]][i]; 
                continue; 
            }
            f[q]=ch[f[u]][i], tag[q]|=tag[f[q]];   
            Q.push(q); 
            if(tag[u])  
                tag[q]=1; 
        }         
    }
}  
void dfs(int u)
{
    if(vis[u]) 
    { 
        printf("TAK
"); 
        exit(0); 
    }
    if(tag[u] || done[u]) return;
    done[u]=vis[u]=1; 
    dfs(ch[u][0]); 
    dfs(ch[u][1]); 
    vis[u]=0; 
}
int main()
{
   //  setIO("input"); 
    int tot; 
    scanf("%d",&tot);
    while(tot--) scanf("%s",str+1), ins(str);  
    build(); 
    dfs(0); 
    printf("NIE
"); 
    return 0;  
}