[FROM WOJ]#3232 [POI2000] 病毒

#3232 [POI2000] 病毒

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

SOL
首先还是将模式串插入到一棵Trie树上。
我们需要考虑什么情况下会出现无限长串，实际上就是要循环。
可以考虑DFS一下，如果能走成环，也许就可以。
但是显然不是所有环都可行，这个环不能包括病毒串。
所以如果一个点是某个病毒串的尾部，就不能走这个点。
Besides，有些时候你的失配指针可能刚好跳回一个病毒串的尾部，所以考虑这种情况。
Last but not the least，DFS时有个很坑的地方，可以把你卡超时！
即使你用vis标记过，依然不能避免有些点被访问到（例如下图），所以还要开一个数组记录这种情况。

代码：

#include<bits/stdc++.h>
#define N 100500
using namespace std;
struct ACtree{
	int ch[N][2],f[N],flag[N],val[N],ans,cnt;
	bool vis[N],b[N];
	inline int idx(char c){return c-'0';}
	inline void insert(char *s){
		int len=strlen(s),u=0;
		for(int i=0;i<len;++i){
			int c=idx(s[i]);
			if(!ch[u][c])ch[u][c]=(++cnt);
			u=ch[u][c];
		}
		flag[u]=1;
	}
	inline void getfail(){
		queue<int>q;f[0]=0;
		for(int register i=0;i<2;i++)if(ch[0][i])f[ch[0][i]]=0,q.push(ch[0][i]);
		while(!q.empty()){
			int u=q.front();q.pop();
			for(int register i=0;i<2;i++){
				if(!ch[u][i]){ch[u][i]=ch[f[u]][i];continue;}
				q.push(ch[u][i]);f[ch[u][i]]=ch[f[u]][i];
				flag[ch[u][i]]|=flag[f[ch[u][i]]];//判断有没有经过完整字串 
			}
		}
	}
	inline bool dfs(int u){
		vis[u]=1;
		for(int register i=0;i<2;i++){
			int v=ch[u][i];
			if(vis[v])return 1;
			if(b[v]||flag[v])continue;
			b[v]=1;//有些时候会走重 
			if(dfs(v))return 1;
		}vis[u]=0;
		return 0;
	}
}ac;
int n;
char s[30005];
int main(){
	scanf("%d",&n);
	for(int register i=1;i<=n;i++)scanf("%s",s),ac.insert(s);
	ac.getfail();
	if(ac.dfs(0))puts("TAK");else puts("NIE");
	return 0;
}