BZOJ 2938 病毒 AC自动机

题目描述

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

示例：

例如如果{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

题解：如果存在一个无线的非病毒的串的话，那么必然可以在trid 树上无线匹配而不匹配到单词节点，所以必然存在环，那么就直接dfs判断即可。

总结：做题时要提炼出问题的本质，仔细分析题目要求的必要条件，从必要条件入手，利用算法。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <queue>
#define N 300000
using namespace std;
int n,st;
queue<int> q;
char str[N];
int vis[N],date[N][2],fail[N],cnt[N],ins[N],tot=0;
void build()
{
	for(int i=0;i<2;i++)
	   if(date[0][i])
	      q.push(date[0][i]);
	while(!q.empty())
	{
		int re=q.front();q.pop();
		for(int i=0;i<2;i++)
		{
			if(!date[re][i])
			{
				date[re][i]=date[fail[re]][i];
				continue;
			}
			q.push(date[re][i]);
			fail[date[re][i]]=date[fail[re]][i];
			cnt[date[re][i]]|=cnt[date[fail[re]][i]];
		}
	}
}
bool dfs(int x)
{
	vis[x]=ins[x]=1;
	for(int i=0;i<2;i++)
	    if(ins[date[x][i]] || (!cnt[date[x][i]] && !vis[date[x][i]] && dfs(date[x][i])))
	       return 1;
	ins[x]=0;
	return 0;
}
int main()
{
	scanf("%d",&n);
	while(n--)
	{
		scanf("%s",str+1);st=strlen(str+1);
		int j=0;
		for(int i=1;i<=st;i++)
		{
			if(!date[j][str[i]-'0']) date[j][str[i]-'0']=++tot;
			j=date[j][str[i]-'0'];
		}
		cnt[j]=1;
	}
	build();
    printf("%s\n" , dfs(0) ? "TAK" : "NIE");
	return 0;
}