#3232 [POI2000] 病毒 【ac自动机+树上判环】

传送门

---

SOL

模拟字符串在ac自动机上匹配的过程
找一个没有不含完整字符串的环，有的话就TAK，否则NIE

但是，这题全是细节。。

注意走过的节点都是生成字符串中的元素。

重点在判断走出的环中有没有完整的单词

1.如果当前是单词节点不行

2.如果它的las数组存在，不行（从它向上的一段是一个完整的单词）
（las：沿着fail跳的一路上，长度最大的 是完整单词节点 的 点 的 编号）

//或者不维护las数组，加一句：

if(flag[fail[u]])flag[u]=1;
//如果当前节点包含完整的子段， 等价于 当前节点是终点 

找环：

维护两个标记：

1.instk：在递归栈中

2.vis：之前被访问过了（防止重复访问）

---

CODE

#include<bits/stdc++.h>
using namespace std;
#define sf scanf
#define pf printf
#define gc getchar()
#define in red()
inline int red(){
	int num=0,f=1;char c=gc;
	for(;!isdigit(c);c=gc)if(c=='-')f=-1;
	for(;isdigit(c);c=gc)num=(num<<1)+(num<<3)+(c^48);
	return num*f;
}
const int N=3e4+10;
char s[N];
int ch[N][2],fail[N],cnt=0,flag[N],instk[N],vis[N];
inline void insert(){
	int len=strlen(s),u=0,c;
	for(int i=0;i<len;++i){
		c=s[i]-'0';
		if(!ch[u][c])ch[u][c]=++cnt;
		u=ch[u][c];
	}
	flag[u]=1;	
}
inline void getfail(){
	queue <int> q;
	for(int i=0;i<2;++i){
		if(ch[0][i])q.push(ch[0][i]);
	}
	while(!q.empty()){
		int u=q.front();q.pop();
		for(int i=0;i<=1;++i){
			if(!ch[u][i]){ch[u][i]=ch[fail[u]][i];continue;}
			if(flag[u])flag[ch[u][i]]=1;
			fail[ch[u][i]]=ch[fail[u]][i]; 
			if(flag[fail[u]])flag[u]=1;
				//判断当前节点包含的合法段 等价于 当前节点是终点 
			q.push(ch[u][i]);
		}
	}
}
inline void dfs(int u){
	if(flag[u])return;
	instk[u]=vis[u]=1;
	for(int i=0;i<=1;++i){
		if(!vis[ch[u][i]])dfs(ch[u][i]);
		else if(instk[ch[u][i]]){
			cout<<"TAK\n";
			exit(0);
		}
	}		
	instk[u]=0;
}
signed main(){ 	
	int n;sf("%d",&n);
	while(n--)sf("%s",s),insert();
	getfail();
	dfs(0);
	cout<<"NIE";
	return 0;
}