BZOJ2938 [Poi2000]病毒【AC自动机】

题目

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

示例：

例如如果{011, 11, 00000}为病毒代码段，那么一个可能的无限长安全代码就是010101…。如果{01, 11, 000000}为病毒代码段，那么就不存在一个无限长的安全代码。

任务：

请写一个程序：

l 读入病毒代码；

l 判断是否存在一个无限长的安全代码；

l 将结果输出

输入格式

第一行包括一个整数n，表示病毒代码段的数目。以下的n行每一行都包括一个非空的01字符串——就是一个病毒代码段。所有病毒代码段的总长度不超过30000。

输出格式

你应在在文本文件WIN.OUT的第一行输出一个单词：

l TAK——假如存在这样的代码；

l NIE——如果不存在。

输入样例

00000

输出样例

NIE

题解

我们对所有串建出一个ACM，我们要做的就是在ACM上找一个路径，使得路径包含环且不包含结束节点

对于所有fail指针指向的节点，如果形成串，那么该节点也不能选，因为fail指针指向的为当前串的前缀

建完机后跑一跑dfs检查环就好了

#include<iostream>

#include<cmath>

#include<queue>

#include<cstdio>

#include<cstring>

#include<algorithm>

#define LL long long int

#define REP(i,n) for (int i = 1; i <= (n); i++)

#define Redge(u) for (int k = h[u],to; k; k = ed[k].nxt)

#define BUG(s,n) for (int i = 1; i <= (n); i++) cout<<s[i]<<' '; puts("");

using namespace std;

const int maxn = 300005,maxm = 1000005,INF = 1000000000;

inline int read(){

	int out = 0,flag = 1; char c = getchar();

	while (c < 48 || c > 57) {if (c == '-') flag = -1; c = getchar();}

	while (c >= 48 && c <= 57) {out = (out << 3) + (out << 1) + c - '0'; c = getchar();}

	return out * flag;

}

int ch[maxn][2],siz,fail[maxn],val[maxn];

char s[maxn];

void insert(char* P){

	int u = 0,len = strlen(P),x;

	for (int i = 0; i < len; i++){

		x = P[i] - '0';

		if (!ch[u][x]) ch[u][x] = ++siz;

		u = ch[u][x];

	}

	val[u] = 1;

}

void getfail(){

	queue<int> q; fail[0] = -1;

	for (int i = 0; i < 2; i++) if (ch[0][i]) q.push(ch[0][i]);

	int u;

	while (!q.empty()){

		u = q.front(); q.pop();

		for (int i = 0; i < 2; i++){

			int v = ch[u][i];

			if (!v) {ch[u][i] = ch[fail[u]][i]; continue;}

			int k = fail[u];

			fail[v] = ch[k][i];

			val[v] |= val[ch[k][i]];

			q.push(v);

		}

	}

}

bool flag;

int vis[maxn],ins[maxn];

bool dfs(int u){

	ins[u] = true;

	for (int i = 0; i < 2; i++){

		int v = ch[u][i];

		if (ins[v]) return true;

		if (vis[v] || val[v]) continue;

		vis[v] = true;

		if (dfs(v)) return true;

	}

	ins[u] = false;

	return false;

}

int main(){

	int n = read();

	while (n--) scanf("%s",s),insert(s);

	getfail();

	dfs(0) ? puts("TAK") : puts("NIE");

	return 0;

}