BZOJ2938[Poi2000]病毒——AC自动机

时间:2024-04-06 22:06:06

题目描述

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

输入

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

输出

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

样例输入

3
01
11
00000

样例输出

NIE
AC自动机经典题,题目要求就是要在trie树上在不走到危险节点的情况下,走出一个环。危险节点就是每个病毒字符串的终止节点,如果一个节点的失配标记指向危险节点,那它自己也是危险节点。在建立trie树并找到每个节点的失配节点(但不要忘了标记危险节点)后只要dfs一下AC自动机,每走到一个点打*问标记,回溯到这个点再把标记删除,如果走到了打标记的点就说明出现了环。
最后附上代码。
#include<cmath>
#include<queue>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
struct tree
{
int fail;
int vis[3];
bool end;
}a[30001];
int cnt;
int n;
int num;
char s[30001];
bool f[30001];
bool v[30001];
void build(char *s)
{
int l=strlen(s);
int now=0;
for(int i=0;i<l;i++)
{
int x=(s[i]-'0');
if(a[now].vis[x]==0)
{
a[now].vis[x]=++cnt;
}
now=a[now].vis[x];
}
a[now].end=true;
}
void bfs()
{
queue<int>q;
for(int i=0;i<2;i++)
{
if(a[0].vis[i]!=0)
{
a[a[0].vis[i]].fail=0;
q.push(a[0].vis[i]);
}
}
while(!q.empty())
{
int now=q.front();
q.pop();
for(int i=0;i<2;i++)
{
if(a[now].vis[i]!=0)
{
a[a[now].vis[i]].fail=a[a[now].fail].vis[i];
q.push(a[now].vis[i]);
if(a[a[a[now].fail].vis[i]].end)
{
a[a[now].vis[i]].end=true;
}
}
else
{
a[now].vis[i]=a[a[now].fail].vis[i];
}
}
}
}
void dfs(int d)
{
v[d]=true;
for(int i=0;i<2;i++)
{
if(v[a[d].vis[i]])
{
printf("TAK");
exit(0);
}
else if(!a[a[d].vis[i]].end&&!f[a[d].vis[i]])
{
f[a[d].vis[i]]=true;
dfs(a[d].vis[i]);
}
}
v[d]=false;
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%s",&s);
build(s);
}
a[0].fail=0;
bfs();
dfs(0);
printf("NIE");
return 0;
}