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

题目描述

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

示例：

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

任务：

请写一个程序：

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

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

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

输入输出格式

输入格式：

在文本文件WIR.IN的第一行包括一个整数n(n≤2000)(n\le 2000)(n≤2000)，表示病毒代码段的数目。以下的n行每一行都包括一个非空的01字符串——就是一个病毒代码段。所有病毒代码段的总长度不超过30000。

输出格式：

在文本文件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\n");
        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\n");
    return 0;
}