【bzoj2938】[Poi2000]病毒 trie图+dfs

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

HINT

Source

听说trie图就是AC自动机改一下？

题意可以理解成 找一个无限长的串使它可以在trie图上无限走，让它永远失配。

那么单词结尾节点就不能走，指向它的fail指针的那个点也不能走，所以干脆就合并起来，这就有trie图了。

这样就形成了一个有向图，现在要从根节点出发找一个环，使得路径上不经过单词结尾节点…dfs即可

原来dfs找环可以加个数组记忆化一下啊…就是判重

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;

const int SZ = 1000010;

int ch[SZ][3],sz = 0,val[SZ];

void insert(char s[])
{
int p = 0;
int l = strlen(s);
for(int i = 0;i < l;i ++)
{
int c = s[i] - '0';
if(!ch[p][c]) ch[p][c] = ++ sz;
p = ch[p][c];
}
val[p] ++;
}

int fail[SZ];

queue<int> q;

void build_ac()
{
fail[0] = 0;
for(int i = 0;i < 2;i ++)
{
int u = ch[0][i];
if(u) { q.push(u); fail[u] = 0; }
}
while(q.size())
{
int f = q.front(); q.pop();
for(int i = 0;i < 2;i ++)
{
int u = ch[f][i];
if(!u) { ch[f][i] = ch[fail[f]][i]; continue; }
q.push(u);
int v = fail[f];
while(v && !ch[v][i]) v = fail[v];
fail[u] = ch[v][i];
val[u] |= val[fail[u]];
}
}
}

bool ins[SZ],vis[SZ];

bool dfs(int u)
{
ins[u] = 1;
for(int i = 0;i < 2;i ++)
{
int v = ch[u][i];
if(ins[v]) return true;
if(val[v] || vis[v]) continue;
vis[v] = 1;
if(dfs(v)) return true;
}
ins[u] = 0;
return false;
}

char s[SZ];
int main()
{
int n;
scanf("%d",&n);
for(int i = 1;i <= n;i ++)
{
scanf("%s",s);
insert(s);
}
build_ac();
if(dfs(0)) puts("TAK");
else puts("NIE");
return 0;
}
/*
3
011
11
00000
*/