bzoj 2938 病毒 AC自动机

#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#define maxn 100005
using namespace std;
char s[maxn];
struct Trie{
bool val[maxn];
int fail[maxn];
int ch[maxn][2],cnt;
void insert()
{
int p=0;
int n=strlen(s+1);
for(int i=1;i<=n;i++)
{
int c=s[i]-'0';
if(!ch[p][c]) ch[p][c]=++cnt;
p=ch[p][c];
}
val[p]=1;
}
queue<int> Q;
void build()
{
for(int i=0;i<=1;i++)
if(ch[0][i]) Q.push(ch[0][i]);
while(!Q.empty())
{
int r=Q.front();Q.pop();
for(int i=0;i<=1;i++)
{
int u=ch[r][i];
if(!u)
{
ch[r][i]=ch[fail[r]][i];
continue;
}
Q.push(u);
int p=fail[r];
while(p&&!ch[p][i]) p=fail[p];
fail[u]=ch[p][i];
val[u]|=val[fail[u]];
}
}
}
bool ins[maxn];
bool vis[maxn];
bool dfs(int x)
{
bool can=0;
ins[x]=vis[x]=1;
for(int i=0;i<=1;i++)
{
int v=ch[x][i];
if(!val[v])
{
if(vis[v])return true;
if(!ins[v]&&dfs(v))return true;
}
}
vis[x]=0;
return false;
}
}AC;
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%s",s+1);
AC.insert();
}
AC.build();
if(AC.dfs(0)) puts("TAK");
else puts("NIE");
return 0;
}