【AC自动机】【bzoj 2938】: [Poi2000]病毒

最新推荐文章于 2018-10-23 20:36:00 发布

willinglive

最新推荐文章于 2018-10-23 20:36:00 发布

阅读量577

点赞数

分类专栏： string 代码

本文链接：https://blog.csdn.net/willinglive/article/details/42100639

版权

代码同时被 2 个专栏收录

22 篇文章 0 订阅

订阅专栏

string

21 篇文章 0 订阅

订阅专栏

http://www.lydsy.com/JudgeOnline/problem.php?id=2938

自从zky发了题解之后，这题做的人蛮多的

我也来写一写

就是建图之后再判环就可以了

不加vis[]也对，但不能保证时间复杂度，会TLE。。。

//#define _TEST _TEST
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <cmath>
#include <algorithm>
#include <queue>
using namespace std;
/************************************************
Code By willinglive    Blog:http://willinglive.cf
************************************************/
#define rep(i,l,r) for(int i=(l),___t=(r);i<=___t;i++)
#define per(i,r,l) for(int i=(r),___t=(l);i>=___t;i--)
#define MS(arr,x) memset(arr,x,sizeof(arr))
#define LL long long
#define INE(i,u,e) for(int i=head[u];~i;i=e[i].next)
inline const int read()
{int r=0,k=1;char c=getchar();for(;c<'0'||c>'9';c=getchar())if(c=='-')k=-1;
for(;c>='0'&&c<='9';c=getchar())r=r*10+c-'0';return k*r;}
/
int n;
char str[30003];
struct node{node *next[2],*fail;bool isword;int id;
void init(){next[0]=next[1]=fail=0;isword=0;}}*root,T[30000];
int cntNode;
bool ins[30003];
bool vis[30003];
/
node* newNode()
{
	T[++cntNode].init();
	T[cntNode].id=cntNode;
	return &T[cntNode];
}
void insert(char *p)
{
	node *o=root;
	for(;*p;p++)
	{
		int son=*p-'0';
		if(o->next[son]==0) o->next[son]=newNode();
		o=o->next[son];
	}
	o->isword=1;
}
void ACMatch()
{
	static queue<node*>q;
	q.push(root); root->fail=0;
	while(!q.empty())
	{
		node *o=q.front(); q.pop();
		rep(i,0,1)
		    if(o->next[i])
		    {
		    	if(o==root) o->next[i]->fail=root;
		    	else
		    	{
		    		o->next[i]->fail=o->fail->next[i];
		    		if(o->next[i]->fail->isword) o->next[i]->isword=1;
		    	}
		    	q.push(o->next[i]);
		    }
		    else
		    {
		    	if(o==root) o->next[i]=root;
		    	else
		    	{
		    		o->next[i]=o->fail->next[i];
		    	}
		    }
	}
}
bool flag=0;
void dfs(node *u)
{
	if(flag) return;
	if(vis[u->id]) return;
	vis[u->id]=1;
	ins[u->id]=1;
	rep(i,0,1)
	{
		node *v=u->next[i];
		if(v->isword) continue;
		if(ins[v->id]) {flag=1;return;}
		dfs(v);
	}
	ins[u->id]=0;
}
/
void input()
{
    n=read();
    root=newNode();
    rep(i,1,n)
    {
    	scanf("%s",str);
    	insert(str);
    }
    
}
void solve()
{
	ACMatch();
	dfs(root);
	if(flag) puts("TAK");
	else puts("NIE");
}
/
int main()
{
    #ifndef _TEST
    freopen("std.in","r",stdin); freopen("std.out","w",stdout);
    #endif
    input(),solve();
    return 0;
}